Made some changes

This commit is contained in:
Nazar Sturanec
2024-05-25 03:26:39 +03:00
parent f8b8747f5e
commit 98c604ff93
3 changed files with 5 additions and 9 deletions
Binary file not shown.
Binary file not shown.
+5 -9
View File
@@ -5,23 +5,19 @@ import train
import preprocessing
import datetime
p = False
start_file = False
while True:
for foldername, subfolders, filenames in os.walk(os.path.join('datasets', 'zombobase-current')):
p = True
start_file = True
for filename in filenames:
if p == True:
time.sleep(3)
print('8' * 8)
if start_file == True:
time.sleep(5)
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
start_file = False