Update collecting-data-for-larger-fpv-model.py

lzh0 · May 27, 2017 · e068e2d · e068e2d
1 parent b42bc9c
commit e068e2d
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/collecting-data-for-larger-fpv-model.py b/collecting-data-for-larger-fpv-model.py
@@ -67,14 +67,18 @@ def keys_to_output(keys):
     return output
 
 
-file_name = 'training_data.npy'
-
-if os.path.isfile(file_name):
-    print('File exists, loading previous data!')
-    training_data = list(np.load(file_name))
-else:
-    print('File does not exist, starting fresh!')
-    training_data = []
+
+starting_value = 486
+
+while True:
+    file_name = 'training_data-{}.npy'.format(starting_value)
+
+    if os.path.isfile(file_name):
+        print('File exists, moving along',starting_value)
+        starting_value += 1
+    else:
+        print('File does not exist, starting fresh!',starting_value)
+        break
 
 
 def main():