Add check.py

check.py

@@ -0,0 +1,27 @@
+import os
+import time
+import shutil
+import train
+import preprocessing
+import datetime
+
+p = False
+while True:
+    for foldername, subfolders, filenames in os.walk(os.path.join('datasets', 'zombobase-current')):
+        p = True
+        for filename in filenames:
+            if p == True:
+                time.sleep(3)
+                print('8' * 8)
+                preprocessing.main(os.path.join('datasets', 'zombobase-current'),'labels', 'images')
+
+                time.sleep(3)
+                train.main()
+
+                time.sleep(3)
+
+                shutil.rmtree(os.path.join('datasets', 'zombobase-current', 'images'))
+                shutil.rmtree(os.path.join('datasets', 'zombobase-current', 'labels'))
+
+                p = False
+