Commit Graph

23 Commits

Author SHA1 Message Date
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
248d54ca03 Update metrics names (#85) 2022-12-24 02:32:24 +01:00
6432afc5f9 Rename img_size to imgsz (#86) 2022-12-24 00:39:09 +01:00
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
7690cae2fc add a naive DDP for model interface (#78)
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-12-19 13:24:39 +08:00
e629335f6d Revert augment_hyps (#70) 2022-12-09 15:49:53 +05:30
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
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
e6737f1207 Predictor support (#65)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-07 10:33:10 +05:30
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
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
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
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
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
512a225ce8 Cli support (#50)
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-23 11:37:27 +01:00
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
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
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
db1031a1a9 Allow setting model attributes before training (#45) 2022-11-17 10:35:53 +05:30
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
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
d143ac666f [rename] - preprocess-batch -> preprocess, preprocess_preds -> postprocess (#42) 2022-11-09 17:35:15 +01:00
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