mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 11:26:36 +00:00
Oleksandr Bezdieniezhnykh
28 lines
744 B
Python
28 lines
744 B
Python
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
|
|
|