Commit Graph

176 Commits (e5cb35edfc3bbc9d7d7db8a6042778a751f0e39e)

Author SHA1 Message Date
Glenn Jocher 0d182e80f1
Add `max_dim==2` argument to `check_imgsz()` () 2 years ago
Glenn Jocher fa8811dcee
`ultralytics 8.0.26` new YOLOv5u models () 2 years ago
Adrian Boguszewski b83374b42d
Use new OpenVINO API instead of deprecated () 2 years ago
Glenn Jocher 522f1937ed
ImageNet names, classify inference, resume fixes () 2 years ago
Glenn Jocher aecd17d455
`ultralytics 8.0.24` mosaic, DDP, download fixes () 2 years ago
Glenn Jocher 899abe9f82
`imgsz` warning fix, download function consolidation () 2 years ago
Ayush Chaurasia 0609561549
Support prediction of list of sources, in-memory dataset and other improvements () 2 years ago
Glenn Jocher a5410ed79e
DDP, Comet, URLError fixes, improved error handling () 2 years ago
Glenn Jocher 6c44ce21d9
`ultralytics 8.0.21` Windows, segments, YAML fixes () 2 years ago
Glenn Jocher 59d4335664
New `guess_model_task()` function () 2 years ago
Glenn Jocher 520825c4b2
`ultralytics 8.0.19` seg/det dataset warning and DDP-cls/seg fixes () 2 years ago
Ayush Chaurasia 936414c615
`ultralytics 8.0.18` new python callbacks and minor fixes () 2 years ago
Glenn Jocher d6a4ffb778
Remove GitPython dependency () 2 years ago
Glenn Jocher 3c4de102f6
8.0.15 CLI float bug fix () 2 years ago
Glenn Jocher 21b701c4ea
Threadpool fixes and CLI improvements () 2 years ago
Glenn Jocher d9a0fba251
`ultralytics 8.0.14` Hydra removal fixes and cleanup () 2 years ago
Glenn Jocher c5fccc3fc4
`ultralytics 8.0.12` - Hydra removal () 2 years ago
Glenn Jocher 6eec39162a
`ultralytics 8.0.11` bug fixes and performance improvements () 2 years ago
Glenn Jocher a86218b767
Fix model re-fuse() in inference loops () 2 years ago
Glenn Jocher 453b5f259a
CLI Simplification () 2 years ago
Ayush Chaurasia 454191bd4b
Fix predict bug on show=True () 2 years ago
Ayush Chaurasia c6985da9de
New YOLOv8 `Results()` class for prediction outputs () 2 years ago
Glenn Jocher 583eac0e80
General refactoring and improvements () 2 years ago
Glenn Jocher f8e32c4c13
General `ultralytics==8.0.6` updates () 2 years ago
Laughing c42e44a021
Release 8.0.5 PR () 2 years ago
Glenn Jocher d0b616e41e
Update README.md () 2 years ago
Glenn Jocher e18ae9d8e1
README and Docs updates with A100 TensorRT times () 2 years ago
Ayush Chaurasia 216cf2ddb6
Release 8.0.4 fixes () 2 years ago
Glenn Jocher 88c9418087
is_docker() fix for `show=True` predict bug () 2 years ago
Glenn Jocher 7f9d7142c2
Minor updates and improvements () 2 years ago
Glenn Jocher c7629e93bd
Add YOLOv5 dataset yamls () 2 years ago
Ayush Chaurasia e371e81aa0
Webcam inference fix () 2 years ago
Ayush Chaurasia 341ce0ff3f
Add YOLOv3 and YOLOv5 model *.yaml files () 2 years ago
Glenn Jocher 0a7d86cd26
Detection training zero-mAP fix () 2 years ago
Glenn Jocher 202f7bffa3
TensorRT Export Fix () 2 years ago
Glenn Jocher 422c49d439
Cleanup () 2 years ago
Ayush Chaurasia 6e5638c128
Return processed outputs from predictor () 2 years ago
Laughing 20fee4100c
Minor fixes () 2 years ago
Ayush Chaurasia 9a2f67b3b4
Log lr for param groups () 2 years ago
Ayush Chaurasia d387359f74
[Docs]: Add customization tutorial and address feedback () 2 years ago
Ayush Chaurasia 07eab49c3d
Add Classification model YAML support () 2 years ago
Ayush Chaurasia 0e5a7ae623
General trainer cleanup () 2 years ago
Glenn Jocher f8a13c49a0
Model saved_model export () 2 years ago
Glenn Jocher 314da263c7
Default confidences predict 0.25, val 0.001 () 2 years ago
Glenn Jocher 3cbf3ec455
Global settings typechecking () 2 years ago
Laughing 19334ebb16
Clean validator () 2 years ago
Glenn Jocher 46cb657b64
Merge `model()` and `model.predict()` () 2 years ago
Glenn Jocher 99275814f1
Add AutoBatch from YOLOv5 () 2 years ago
Glenn Jocher 172cef2d20
CoreML NMS and half fixes () 2 years ago
Glenn Jocher 55bdca6768
Improvements () 2 years ago
Ayush Chaurasia dcd8ef68e6
Add colab template and minor fixes () 2 years ago
Glenn Jocher 3c18aa5278
Simplify argument names () 2 years ago
Glenn Jocher d74de2582c
Pip debug fixes () 2 years ago
Laughing 340376f7a6
Fix resume () 2 years ago
Ayush Chaurasia 82c849c163
Fix load and resume and update autodownload endpoint () 2 years ago
Ayush Chaurasia 6d5123297e
Fix CLI detect and segment resume () 2 years ago
Glenn Jocher 8f3cd52844
Fix `yolo mode=train` CLI bug on model load () 2 years ago
Glenn Jocher c3d961fb03
Unified model loading with backwards compatibility () 2 years ago
Glenn Jocher af6e3c536b
Override fixes and general updates () 2 years ago
Glenn Jocher d76d7af566 Simplify usage, remove YOLO.new() and YOLO.load() 2 years ago
Ayush Chaurasia 2bc9a5c87e
HUB setup () 2 years ago
Ayush Chaurasia c6eb6720de
Add pred, export and val callbacks () 2 years ago
Glenn Jocher 63c7a74691
Replace `nosave` and `noval` with `save` and `val` () 2 years ago
Glenn Jocher 598f17a472
Add global `settings.yaml` in `USER_CONFIG_DIR` () 2 years ago
Glenn Jocher a9b9fe7618
Add Dockerfiles and update Docs README () 2 years ago
Glenn Jocher df4fc14c10
Docstring additions () 2 years ago
Glenn Jocher c9f3e469cb
Add CoreML iOS updates () 2 years ago
Glenn Jocher fec13ec773
Add YOLO8x6 and YAML syntax improvements () 2 years ago
Glenn Jocher 076d73cfaa
Create Exporter() Class () 2 years ago
Laughing a9dc1637c2
Mask pycocotools () 2 years ago
Glenn Jocher 37e0f6e0e4
Import DEFAULT_CONFIG from utils/__init__.py () 2 years ago
Glenn Jocher 92dad1c1b5
Start export implementation () 2 years ago
Ayush Chaurasia c1b38428bc
Update save_dir rank check () 2 years ago
Ayush Chaurasia 100e21ee80
fix multiple dirs in DDP mode () 2 years ago
Ayush Chaurasia 34829a6b29
Make YOLO a module () 2 years ago
Ayush Chaurasia 0303ced8ab
Model enhancements 2 () 2 years ago
Ayush Chaurasia 384f0ef1c6
Model interface enhancement () 2 years ago
Glenn Jocher 38d6df55cb
Fix `save_json(predn, batch)` () 2 years ago
Ayush Chaurasia 8028e2b1b8
Make optimizer static method inside trainer () 2 years ago
Laughing 5c6d11bdb2
Allocated updated pycocotools metrics fix () 2 years ago
Glenn Jocher 6f0ba81427
Add best.pt val and COCO pycocotools val () 2 years ago
Glenn Jocher 48cffa176e
Shorten module paths with new 'nn' dir () 2 years ago
Ayush Chaurasia 4fb04be20b
DDP and new dataloader Fix () 2 years ago
Ayush Chaurasia 16e3c08883
Import YOLOv5 dataloader () 2 years ago
Glenn Jocher 249dfbdc05
Fix ClearML Mosaic callback to 'on_train_epoch_end' () 2 years ago
Glenn Jocher 98815d560f
YOLOv5 updates () 2 years ago
Glenn Jocher ebd3cfb2fd
YOLOv8 architecture updates from R&D branch () 2 years ago
Ayush Chaurasia 5fbea25f0b
Use integrations only in master process () 2 years ago
Glenn Jocher cb4f20f3cf
Add TensorBoard support () 2 years ago
Glenn Jocher 248d54ca03
Update metrics names () 2 years ago
Glenn Jocher 6432afc5f9
Rename `img_size` to `imgsz` () 2 years ago
Ayush Chaurasia ae2443c210
Add flops, num_params, inference speed logging and best.pt logging () 2 years ago
Ayush Chaurasia d198f3f9d1
Update docs () 2 years ago
Laughing 7690cae2fc
add a naive DDP for model interface () 2 years ago
Ayush Chaurasia 48c95ba083
Change class depending on dataset in model interface () 2 years ago
Ayush Chaurasia eb5adf4e0b
Model enhancement () 2 years ago
Ayush Chaurasia d85b44f259
Update docs () 2 years ago
Laughing d63ee112d4
General cleanup () 2 years ago
Ayush Chaurasia 7ae45c6cc4
[WIP] Model interface () 2 years ago
Ayush Chaurasia e6737f1207
Predictor support () 2 years ago