19 Commits

Author SHA1 Message Date
Glenn Jocher
98815d560f
YOLOv5 updates (#90)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-25 14:33:18 +01:00
Glenn Jocher
ebd3cfb2fd
YOLOv8 architecture updates from R&D branch (#88)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-24 18:10:44 +01:00
Glenn Jocher
248d54ca03
Update metrics names (#85) 2022-12-24 02:32:24 +01:00
Glenn Jocher
6432afc5f9
Rename img_size to imgsz (#86) 2022-12-24 00:39:09 +01:00
Glenn Jocher
f0fff8c13e
Simplify cli.py and fix Detect train Usage (#83)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-22 15:22:16 +01:00
Laughing
d63ee112d4
General cleanup (#69)
Co-authored-by: ayush chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-08 19:58:13 +05:30
Ayush Chaurasia
7ae45c6cc4
[WIP] Model interface (#68)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
2022-12-08 13:22:56 +05:30
Laughing
fbeeb5d1e1
add resuming (#63)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-06 08:26:41 +05:30
Ayush Chaurasia
7ec7cf3aef
Detection support (#60)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
2022-12-03 21:01:23 +05:30
Ayush Chaurasia
5a52e7663a
standalone val (#56)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-30 15:04:44 +05:30
Laughing
3a241e4cea
update segment training (#57)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: ayush chaurasia <ayush.chaurarsia@gmail.com>
2022-11-29 17:00:08 +05:30
Ayush Chaurasia
298287298d
Add clearml logging (#51)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-24 15:25:03 +01:00
Ayush Chaurasia
4291b9c31c
Add EMA and model checkpointing (#49)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-19 19:07:26 +01:00
Glenn Jocher
27d6545117
General console printout updates (#48)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-19 16:08:16 +01:00
Laughing
47f1cb3ef4
Fix some cuda training issues of segmentation (#46)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-17 18:14:02 +05:30
Ayush Chaurasia
db1031a1a9
Allow setting model attributes before training (#45) 2022-11-17 10:35:53 +05:30
Ayush Chaurasia
832ea56eb4
update model initialization design, supports custom data/num_classes (#44)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-15 15:36:29 +01:00
Ayush Chaurasia
1f3aad86c1
new check_dataset functions (#43)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-12 15:02:41 +01:00
Ayush Chaurasia
f56c9bcc26
Segmentation support & other enchancements (#40)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-08 16:27:57 +01:00