Revert "Made some changes"

This reverts commit 98c604ff93.
This commit is contained in:
Oleksandr Bezdieniezhnykh
2024-05-26 23:10:18 +03:00
parent 98c604ff93
commit a1e08e640f
3 changed files with 9 additions and 5 deletions
Binary file not shown.
Binary file not shown.
+9 -5
View File
@@ -5,19 +5,23 @@ import train
import preprocessing import preprocessing
import datetime import datetime
start_file = False p = False
while True: while True:
for foldername, subfolders, filenames in os.walk(os.path.join('datasets', 'zombobase-current')): for foldername, subfolders, filenames in os.walk(os.path.join('datasets', 'zombobase-current')):
start_file = True p = True
for filename in filenames: for filename in filenames:
if start_file == True: if p == True:
time.sleep(5) time.sleep(3)
print('8' * 8)
preprocessing.main(os.path.join('datasets', 'zombobase-current'),'labels', 'images') preprocessing.main(os.path.join('datasets', 'zombobase-current'),'labels', 'images')
time.sleep(3) time.sleep(3)
train.main() train.main()
time.sleep(3) time.sleep(3)
shutil.rmtree(os.path.join('datasets', 'zombobase-current', 'images')) shutil.rmtree(os.path.join('datasets', 'zombobase-current', 'images'))
shutil.rmtree(os.path.join('datasets', 'zombobase-current', 'labels')) shutil.rmtree(os.path.join('datasets', 'zombobase-current', 'labels'))
start_file = False p = False