Glenn Jocher
|
202f7bffa3
|
TensorRT Export Fix (#171)
|
2 years ago |
|
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>
|
2 years ago |
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>
|
2 years ago |
Laughing
|
20fee4100c
|
Minor fixes (#162)
|
2 years ago |
|
Ayush Chaurasia
|
9a2f67b3b4
|
Log lr for param groups (#159)
|
2 years ago |
|
Ayush Chaurasia
|
d387359f74
|
[Docs]: Add customization tutorial and address feedback (#155)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
Glenn Jocher
|
f8a13c49a0
|
Model saved_model export (#151)
|
2 years ago |
|
Glenn Jocher
|
314da263c7
|
Default confidences predict 0.25, val 0.001 (#149)
|
2 years ago |
|
Glenn Jocher
|
3cbf3ec455
|
Global settings typechecking (#148)
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
Glenn Jocher
|
99275814f1
|
Add AutoBatch from YOLOv5 (#145)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
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>
|
2 years ago |
|
Glenn Jocher
|
55bdca6768
|
Improvements (#142)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
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>
|
2 years ago |
|
Glenn Jocher
|
3c18aa5278
|
Simplify argument names (#141)
|
2 years ago |
|
Glenn Jocher
|
d74de2582c
|
Pip debug fixes (#139)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Laughing
|
340376f7a6
|
Fix resume (#138)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
Glenn Jocher
|
d76d7af566
|
Simplify usage, remove YOLO.new() and YOLO.load()
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
Glenn Jocher
|
598f17a472
|
Add global `settings.yaml` in `USER_CONFIG_DIR` (#125)
|
2 years ago |
|
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>
|
2 years ago |
|
Glenn Jocher
|
df4fc14c10
|
Docstring additions (#122)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
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>
|
2 years ago |
|
Glenn Jocher
|
fec13ec773
|
Add YOLO8x6 and YAML syntax improvements (#120)
|
2 years ago |
|
Glenn Jocher
|
076d73cfaa
|
Create Exporter() Class (#117)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Laughing
|
a9dc1637c2
|
Mask pycocotools (#116)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Glenn Jocher
|
37e0f6e0e4
|
Import DEFAULT_CONFIG from utils/__init__.py (#115)
|
2 years ago |
|
Glenn Jocher
|
92dad1c1b5
|
Start export implementation (#110)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Ayush Chaurasia
|
c1b38428bc
|
Update save_dir rank check (#114)
Co-authored-by: Laughing-q <1185102784@qq.com>
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
Laughing
|
5c6d11bdb2
|
Allocated updated pycocotools metrics fix (#101)
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>
|
2 years ago |
|
Glenn Jocher
|
6f0ba81427
|
Add best.pt val and COCO pycocotools val (#98)
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>
|
2 years ago |
|
Glenn Jocher
|
48cffa176e
|
Shorten module paths with new 'nn' dir (#96)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Ayush Chaurasia
|
4fb04be20b
|
DDP and new dataloader Fix (#95)
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>
|
2 years ago |
|
Ayush Chaurasia
|
16e3c08883
|
Import YOLOv5 dataloader (#94)
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>
|
2 years ago |
