ferdzo/vesselDetection
1
0
Fork 0
mirror of https://github.com/ferdzo/vesselDetection.git synced 2026-04-05 10:16:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added small changes and migrated to Yolo v11

Browse Source
This commit is contained in:
Andrej Mickov
2025-06-26 18:41:51 +02:00
parent d9ecbf413b
commit 5ece085064
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
train.py
View File

@@ -21,6 +21,7 @@ train_params = {
'exist_ok': True, 'exist_ok': True,
'save_period': 2, 'save_period': 2,
'workers': 8, 'workers': 8,
'auto_augment': 'autoaugment',
} }
model.train(data=data_path, **train_params) model.train(data=data_path, **train_params)