azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-23 02:26:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

train.py fix

Browse Source
This commit is contained in:
Alex Bezdieniezhnykh
2024-06-19 03:47:57 +03:00
parent 6c50dd19b7
commit eba8b62db8
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+4 -2
View File
@@ -100,13 +100,15 @@ if __name__ == '__main__':
# form_dataset() # form_dataset()
# create_yaml() # create_yaml()
m = latest_model or 'yolov10x.yaml'
print(f'Initial model: {m}')
model = YOLOv10(latest_model or 'yolov10x.yaml') model = YOLOv10(latest_model or 'yolov10x.yaml')
folder = f'{prefix}2024-06-18' folder = f'{prefix}2024-06-18'
yaml = abspath(path.join('datasets', today_folder, 'data.yaml')) yaml = abspath(path.join('datasets', folder, 'data.yaml'))
results = model.train(data=yaml, epochs=100, batch=10, imgsz=640) results = model.train(data=yaml, epochs=100, batch=10, imgsz=640)
shutil.copy(f'{results.save_dir}/weights/best.pt', latest_model) shutil.copy(f'{results.save_dir}/weights/best.pt', latest_model)
shutil.copytree(results.save_dir, f'models/{today_folder}') shutil.copytree(results.save_dir, f'models/{folder}')
shutil.rmtree('runs') shutil.rmtree('runs')
shutil.rmtree('models/zombobase-latest') shutil.rmtree('models/zombobase-latest')