add dataset-visualiser.py

2026-04-22 08:56:35 +00:00 · 2024-06-19 02:28:50 +03:00
parent 165d609d01
commit 6c50dd19b7
5 changed files with 41 additions and 6 deletions
@@ -102,6 +102,7 @@ if __name__ == '__main__':

    model = YOLOv10(latest_model or 'yolov10x.yaml')

+    folder = f'{prefix}2024-06-18'
    yaml = abspath(path.join('datasets', today_folder, 'data.yaml'))
    results = model.train(data=yaml, epochs=100, batch=10, imgsz=640)