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

refactoring, fixes

Browse Source
This commit is contained in:
zxsanny
2024-08-23 13:47:54 +03:00
parent 9c9e3d8d44
commit 543cead191
1 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+5 -7
View File
@@ -19,7 +19,7 @@ valid_set = 20
test_set = 10
def form_dataset(set_date: datetime):
def form_dataset(from_date: datetime):
makedirs(today_dataset, exist_ok=True)
images = []
with scandir(processed_images_dir) as imd:
@@ -27,9 +27,9 @@ def form_dataset(set_date: datetime):
if not image_file.is_file():
continue
mod_time = datetime.fromtimestamp(image_file.stat().st_mtime)
if set_date is None:
if from_date is None:
images.append(image_file)
elif mod_time > set_date:
elif mod_time > from_date:
images.append(image_file)
print('shuffling images')
@@ -142,9 +142,7 @@ if __name__ == '__main__':
cur_folder = today_dataset
yaml = abspath(path.join(cur_folder, 'data.yaml'))
results = model.train(data=yaml, epochs=100, batch=60, imgsz=640, save_period=1)
results = model.train(data=yaml, epochs=100, batch=60, imgsz=640)
shutil.copy(f'{results.save_dir}/weights/best.pt', latest_model)
shutil.copytree(results.save_dir, path.join(models_dir, cur_folder))
shutil.copytree(results.save_dir, path.join(models_dir, today_folder))
shutil.rmtree('runs')
shutil.rmtree(path.join(models_dir, f'{prefix}latest'))