97 Commits

Author SHA1 Message Date
Glenn Jocher
7f9d7142c2
Minor updates and improvements (#216)
Co-authored-by: Hardik Dava <39372750+hardikdava@users.noreply.github.com>
Co-authored-by: Onuralp Sezer <thunderbirdtr@fedoraproject.org>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-11 00:13:44 +01:00
Glenn Jocher
c7629e93bd
Add YOLOv5 dataset yamls (#207)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-10 18:33:44 +01:00
Ayush Chaurasia
e371e81aa0
Webcam inference fix (#202)
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-10 18:29:11 +01:00
Ayush Chaurasia
341ce0ff3f
Add YOLOv3 and YOLOv5 model *.yaml files (#169)
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-10 12:33:28 +01:00
Glenn Jocher
0a7d86cd26
Detection training zero-mAP fix (#192) 2023-01-10 12:18:39 +01:00
Glenn Jocher
202f7bffa3
TensorRT Export Fix (#171) 2023-01-10 01:00:44 +01:00
Glenn Jocher
422c49d439
Cleanup (#168)
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-09 23:22:33 +01:00
Ayush Chaurasia
6e5638c128
Return processed outputs from predictor (#161)
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>
Co-authored-by: Laughing-q <1185102784@qq.com>
2023-01-09 19:40:44 +01:00
Laughing
20fee4100c
Minor fixes (#162) 2023-01-09 13:41:34 +01:00
Ayush Chaurasia
9a2f67b3b4
Log lr for param groups (#159) 2023-01-09 00:27:14 +01:00
Ayush Chaurasia
d387359f74
[Docs]: Add customization tutorial and address feedback (#155)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-01-08 14:01:22 +01:00
Ayush Chaurasia
07eab49c3d
Add Classification model YAML support (#154)
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-07 20:04:34 +01:00
Ayush Chaurasia
0e5a7ae623
General trainer cleanup (#147)
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>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
2023-01-07 19:25:48 +05:30
Glenn Jocher
f8a13c49a0
Model saved_model export (#151) 2023-01-06 21:44:44 +01:00
Glenn Jocher
314da263c7
Default confidences predict 0.25, val 0.001 (#149) 2023-01-05 15:56:52 +01:00
Glenn Jocher
3cbf3ec455
Global settings typechecking (#148) 2023-01-05 15:13:22 +01:00
Laughing
19334ebb16
Clean validator (#144)
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
46cb657b64
Merge model() and model.predict() (#146)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-05 14:13:29 +01:00
Glenn Jocher
99275814f1
Add AutoBatch from YOLOv5 (#145)
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 (#143)
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 (#142)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-04 16:37:46 +01:00
Ayush Chaurasia
dcd8ef68e6
Add colab template and minor fixes (#137)
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>
2023-01-03 21:30:39 +01:00
Glenn Jocher
3c18aa5278
Simplify argument names (#141) 2023-01-03 19:47:56 +01:00
Glenn Jocher
d74de2582c
Pip debug fixes (#139)
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 (#138)
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 (#136)
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
Ayush Chaurasia
6d5123297e
Fix CLI detect and segment resume (#134)
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:01:17 +01:00
Glenn Jocher
8f3cd52844
Fix yolo mode=train CLI bug on model load (#133)
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 (#132)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-02 17:37:23 +01:00
Glenn Jocher
af6e3c536b
Override fixes and general updates (#129)
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 (#108)
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 (#126)
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
63c7a74691
Replace nosave and noval with save and val (#127)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-01 17:50:06 +01:00
Glenn Jocher
598f17a472
Add global settings.yaml in USER_CONFIG_DIR (#125) 2022-12-31 21:40:41 +01:00
Glenn Jocher
a9b9fe7618
Add Dockerfiles and update Docs README (#124)
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-31 18:39:42 +01:00
Glenn Jocher
df4fc14c10
Docstring additions (#122)
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 (#121)
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
fec13ec773
Add YOLO8x6 and YAML syntax improvements (#120) 2022-12-30 13:32:08 +01:00
Glenn Jocher
076d73cfaa
Create Exporter() Class (#117)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-30 01:28:41 +01:00
Laughing
a9dc1637c2
Mask pycocotools (#116)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-29 18:01:03 +01:00
Glenn Jocher
37e0f6e0e4
Import DEFAULT_CONFIG from utils/__init__.py (#115) 2022-12-29 14:58:03 +01:00
Glenn Jocher
92dad1c1b5
Start export implementation (#110)
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
c1b38428bc
Update save_dir rank check (#114)
Co-authored-by: Laughing-q <1185102784@qq.com>
2022-12-29 13:02:01 +01:00
Ayush Chaurasia
100e21ee80
fix multiple dirs in DDP mode (#113)
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 (#111)
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
Ayush Chaurasia
0303ced8ab
Model enhancements 2 (#109)
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 21:01:51 +05:30
Ayush Chaurasia
384f0ef1c6
Model interface enhancement (#106)
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
38d6df55cb
Fix save_json(predn, batch) (#105)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-27 18:42:36 +01:00
Ayush Chaurasia
8028e2b1b8
Make optimizer static method inside trainer (#103)
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-27 17:02:03 +01:00