mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 12:06:36 +00:00
Nazar Sturanec
24 lines
746 B
Python
24 lines
746 B
Python
import os
|
|
import time
|
|
import shutil
|
|
import train
|
|
import preprocessing
|
|
import datetime
|
|
|
|
start_file = False
|
|
while True:
|
|
for foldername, subfolders, filenames in os.walk(os.path.join('datasets', 'zombobase-current')):
|
|
start_file = True
|
|
for filename in filenames:
|
|
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'))
|
|
|
|
start_file = False
|
|
|