74 Commits

Author SHA1 Message Date
Glenn Jocher
3cbf3ec455
Global settings typechecking () 2023-01-05 15:13:22 +01:00
Laughing
19334ebb16
Clean validator ()
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>
2023-01-05 14:14:01 +01:00
Glenn Jocher
99275814f1
Add AutoBatch from YOLOv5 ()
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-05 13:49:16 +01:00
Glenn Jocher
172cef2d20
CoreML NMS and half fixes ()
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-05 00:20:54 +01:00
Glenn Jocher
55bdca6768
Improvements ()
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-04 16:37:46 +01:00
Glenn Jocher
d74de2582c
Pip debug fixes ()
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-03 18:54:44 +01:00
Laughing
340376f7a6
Fix resume ()
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-03 18:36:22 +05:30
Ayush Chaurasia
82c849c163
Fix load and resume and update autodownload endpoint ()
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>
2023-01-02 21:12:01 +01:00
Glenn Jocher
c5c86a3acd
CLI DDP fixes () 2023-01-02 19:55:04 +01:00
Glenn Jocher
8f3cd52844
Fix yolo mode=train CLI bug on model load ()
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-02 19:24:44 +01:00
Glenn Jocher
c3d961fb03
Unified model loading with backwards compatibility ()
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-02 17:37:23 +01:00
Ayush Chaurasia
8996c5c6cf
[Docs]: Link buttons, add autobackend, BaseModel and ops ()
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>
2023-01-02 20:42:30 +05:30
Glenn Jocher
af6e3c536b
Override fixes and general updates ()
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Kalen Michael <kalenmike@gmail.com>
2023-01-02 15:55:32 +01:00
Glenn Jocher
d76d7af566 Simplify usage, remove YOLO.new() and YOLO.load() 2023-01-02 15:42:01 +01:00
Ayush Chaurasia
2bc9a5c87e
HUB setup ()
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>
2023-01-01 20:21:14 +01:00
Ayush Chaurasia
c6eb6720de
Add pred, export and val callbacks ()
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>
2023-01-01 18:16:10 +01:00
Glenn Jocher
598f17a472
Add global settings.yaml in USER_CONFIG_DIR () 2022-12-31 21:40:41 +01:00
Glenn Jocher
df4fc14c10
Docstring additions ()
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-31 13:42:45 +01:00
Glenn Jocher
c9f3e469cb
Add CoreML iOS updates ()
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-30 21:33:43 +01:00
Glenn Jocher
076d73cfaa
Create Exporter() Class ()
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-30 01:28:41 +01:00
Glenn Jocher
92dad1c1b5
Start export implementation ()
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-29 14:17:14 +01:00
Ayush Chaurasia
100e21ee80
fix multiple dirs in DDP mode ()
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-29 15:02:17 +05:30
Ayush Chaurasia
34829a6b29
Make YOLO a module ()
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-28 19:38:37 +01:00
Laughing
8406b49b49
Integration of v8 segmentation ()
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-28 16:01:38 +01:00
Ayush Chaurasia
384f0ef1c6
Model interface enhancement ()
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-28 13:35:01 +01:00
Glenn Jocher
6f0ba81427
Add best.pt val and COCO pycocotools val ()
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>
2022-12-27 04:56:24 +01:00
Ayush Chaurasia
a1808eeda4
logger updates ()
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-26 18:12:17 +01:00
Glenn Jocher
48cffa176e
Shorten module paths with new 'nn' dir ()
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-26 17:34:28 +01:00
Ayush Chaurasia
4fb04be20b
DDP and new dataloader Fix ()
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-26 15:35:49 +01:00
Ayush Chaurasia
16e3c08883
Import YOLOv5 dataloader ()
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-26 01:30:01 +01:00
Glenn Jocher
ae05d44877
Simplify augmentations ()
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-25 21:21:26 +01:00
Glenn Jocher
249dfbdc05
Fix ClearML Mosaic callback to 'on_train_epoch_end' () 2022-12-25 17:04:24 +01:00
Ayush Chaurasia
0298821467
ClearML: Mosaic Logging ()
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-25 14:36:41 +01:00
Glenn Jocher
98815d560f
YOLOv5 updates ()
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 ()
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
cb4f20f3cf
Add TensorBoard support ()
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-24 14:37:46 +01:00
Glenn Jocher
248d54ca03
Update metrics names () 2022-12-24 02:32:24 +01:00
Glenn Jocher
6432afc5f9
Rename img_size to imgsz () 2022-12-24 00:39:09 +01:00
Ayush Chaurasia
ae2443c210
Add flops, num_params, inference speed logging and best.pt logging ()
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-23 20:14:21 +01:00
Ayush Chaurasia
40f7e0806e
Add v8 modules ()
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-21 00:29:58 +01:00
Ayush Chaurasia
681cfc1c35
Make config overrides user friendly ()
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-20 09:50:08 +05:30
Laughing
7690cae2fc
add a naive DDP for model interface ()
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
Ayush Chaurasia
eb5adf4e0b
Model enhancement ()
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-14 14:33:31 +05:30
Laughing
e629335f6d
Revert augment_hyps () 2022-12-09 15:49:53 +05:30
Laughing
d63ee112d4
General cleanup ()
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
e6737f1207
Predictor support ()
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
Laughing
fbeeb5d1e1
add resuming ()
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 ()
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 ()
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 ()
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