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

small refactoring

Browse Source
This commit is contained in:
Oleksandr Bezdieniezhnykh
2024-06-08 19:34:17 +03:00
parent 56017f2c23
commit a9b777acc4
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+2 -1
View File
@@ -4,8 +4,9 @@ from datetime import datetime
from pathlib import Path from pathlib import Path
from ultralytics import YOLOv10 from ultralytics import YOLOv10
from constants import current_images_dir, current_labels_dir, annotation_classes, today_dataset, prefix from constants import current_images_dir, current_labels_dir, annotation_classes, prefix
today_dataset = os.path.join('datasets', f'{prefix}{datetime.now():%Y-%m-%d}')
yaml_path = os.path.join(today_dataset, 'data.yaml') yaml_path = os.path.join(today_dataset, 'data.yaml')
train_set = 70 train_set = 70
valid_set = 20 valid_set = 20