153 Commits

Author SHA1 Message Date
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
a8a70390ab Created using Colaboratory 2023-01-10 11:29:26 +01:00
Glenn Jocher
3cc63950f3
Enable Docker deployment in docker.yaml (#184) 2023-01-10 11:00:27 +01:00
Glenn Jocher
0755ede91a
Update notebooks (#178) 2023-01-10 10:26:31 +01:00
Glenn Jocher
b206a49207 Created using Colaboratory 2023-01-10 10:24:43 +01:00
Glenn Jocher
c0c0c138c1
Create cla.yml (#175) 2023-01-10 09:55:25 +01:00
Glenn Jocher
2866003ecf
Add export to CI (#174) 2023-01-10 09:52:48 +01:00
Glenn Jocher
f80ff923e7
Start Multi-OS CI (#172) 2023-01-10 08:23:32 +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
Glenn Jocher
6feba17760 Created using Colaboratory 2023-01-09 20:27:41 +01:00
Glenn Jocher
006a5f02ba Created using Colaboratory 2023-01-09 20:25:57 +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
Ayush Chaurasia
cb4801888e
README and Docs updates (#166)
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-09 19:30:42 +01:00
Laughing
5e290e0d28
fix dataset (#167) 2023-01-09 21:55:20 +05:30
Glenn Jocher
b84ed678de Created using Colaboratory 2023-01-09 15:49:12 +01:00
Glenn Jocher
dd61b150f2 Created using Colaboratory 2023-01-09 15:47:13 +01:00
Ayush Chaurasia
449c89e5dc
Map ultralytics and yolo CLI commands (#165)
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-09 15:31:16 +01:00
Glenn Jocher
96fbf9ce58
Update docs with YOLOv8 banner (#160)
Co-authored-by: Paula Derrenger <107626595+pderrenger@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-09 15:24:01 +01:00
Glenn Jocher
fdf294e4e8
Update CONTRIBUTING.md 2023-01-09 14:53:24 +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
Glenn Jocher
e79ea1666c
Fix Classification train logging (#157)
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>
2023-01-08 17:52:44 +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
Laughing
c985eaba0d
Fix segment coco val (#156)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-08 13:51:10 +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
Ayush Chaurasia
d17d1e064d
Add Comet integration (#153)
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 21:08:57 +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
c5c86a3acd
CLI DDP fixes (#135) 2023-01-02 19:55:04 +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
Ayush Chaurasia
8996c5c6cf
[Docs]: Link buttons, add autobackend, BaseModel and ops (#130)
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 (#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
7b21a87da3
HUB initial 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:37 +01:00