Commit Graph
83 Commits
Author SHA1 Message Date
Glenn Jocher 03f56791a3 Update YOLOv5 to v8 comments 2023-01-12 01:37:22 +01:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
e18ae9d8e1 README and Docs updates with A100 TensorRT times (#270)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-11 21:54:41 +01:00
Glenn JocherandGitHub 00c57f2c97 Add YOLO comparison plots (#260) 2023-01-11 14:25:21 +01:00
Glenn JocherGitHubHardik DavaOnuralp Sezerpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
88c9418087 is_docker() fix for show=True predict bug (#218)
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:37:07 +01:00
Glenn JocherGitHubHardik DavaOnuralp Sezerpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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 JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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
c8e3c5db4b Cleanup (#197)
Co-authored-by: Onuralp Sezer <thunderbirdtr@gmail.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
2023-01-10 13:31:43 +01:00
Glenn Jocher 840c35a0aa Created using Colaboratory 2023-01-10 13:10:26 +01:00
Glenn Jocher 1135701163 Created using Colaboratory 2023-01-10 12:52:30 +01:00
Glenn JocherandGitHub 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 JocherandGitHub 3cc63950f3 Enable Docker deployment in docker.yaml (#184) 2023-01-10 11:00:27 +01:00
Glenn JocherandGitHub 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 JocherandGitHub c0c0c138c1 Create cla.yml (#175) 2023-01-10 09:55:25 +01:00
Glenn JocherandGitHub 2866003ecf Add export to CI (#174) 2023-01-10 09:52:48 +01:00
Glenn JocherandGitHub f80ff923e7 Start Multi-OS CI (#172) 2023-01-10 08:23:32 +01:00
Glenn JocherandGitHub 202f7bffa3 TensorRT Export Fix (#171) 2023-01-10 01:00:44 +01:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Kalen Michael
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
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
Glenn JocherGitHubPaula Derrengerpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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 JocherandGitHub fdf294e4e8 Update CONTRIBUTING.md 2023-01-09 14:53:24 +01:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Ayush Chaurasia
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
Glenn JocherandGitHub f8a13c49a0 Model saved_model export (#151) 2023-01-06 21:44:44 +01:00
Glenn JocherandGitHub 314da263c7 Default confidences predict 0.25, val 0.001 (#149) 2023-01-05 15:56:52 +01:00
Glenn JocherandGitHub 3cbf3ec455 Global settings typechecking (#148) 2023-01-05 15:13:22 +01:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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 JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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 JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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 JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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
Glenn JocherandGitHub 3c18aa5278 Simplify argument names (#141) 2023-01-03 19:47:56 +01:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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
Glenn JocherandGitHub c5c86a3acd CLI DDP fixes (#135) 2023-01-02 19:55:04 +01:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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 JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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 JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Kalen Michael
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
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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 JocherandGitHub 598f17a472 Add global settings.yaml in USER_CONFIG_DIR (#125) 2022-12-31 21:40:41 +01:00
Glenn JocherGitHubAyush Chaurasiapre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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 JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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 JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Ayush Chaurasia
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 JocherandGitHub fec13ec773 Add YOLO8x6 and YAML syntax improvements (#120) 2022-12-30 13:32:08 +01:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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
Glenn JocherandGitHub 37e0f6e0e4 Import DEFAULT_CONFIG from utils/__init__.py (#115) 2022-12-29 14:58:03 +01:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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
Glenn JocherandGitHub 441126a77e Add l/x models (#100) 2022-12-27 05:06:15 +01:00
Glenn JocherGitHubayush chaurasiapre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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>
2022-12-27 04:56:24 +01:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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>
2022-12-26 17:34:28 +01:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
ae05d44877 Simplify augmentations (#93)
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 JocherandGitHub 249dfbdc05 Fix ClearML Mosaic callback to 'on_train_epoch_end' (#92) 2022-12-25 17:04:24 +01:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
98815d560f YOLOv5 updates (#90)
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 JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
ebd3cfb2fd YOLOv8 architecture updates from R&D branch (#88)
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 JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
cb4f20f3cf Add TensorBoard support (#87)
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 JocherandGitHub 248d54ca03 Update metrics names (#85) 2022-12-24 02:32:24 +01:00
Glenn JocherandGitHub 6432afc5f9 Rename img_size to imgsz (#86) 2022-12-24 00:39:09 +01:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
f0fff8c13e Simplify cli.py and fix Detect train Usage (#83)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-22 15:22:16 +01:00
Glenn JocherandGitHub 74feef30c4 Update cache_version = 1.0 (#82) 2022-12-22 12:44:28 +01:00
Glenn JocherandGitHub 479992093c Flake8 updates (#66) 2022-12-06 15:09:53 -08:00
Glenn JocherandGitHub de3e6ca54d Fix CITATION.cff typos (#64) 2022-12-05 12:17:25 -08:00
Glenn JocherandGitHub 6903ccdf6b Create CITATION.cff (#62) 2022-12-05 12:14:59 -08:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
27d6545117 General console printout updates (#48)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-19 16:08:16 +01:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
8530e3fae0 Logging fix from YOLOv5 (#47)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-19 14:51:52 +01:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
4c68b9dcf6 Add is_colab() and is_kaggle() (#41)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-08 17:14:49 +01:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
523eff99e2 Update ci.yaml (#37)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-07 19:27:55 +01:00
Glenn JocherandGitHub 06dd49d862 Create dependabot.yml (#38) 2022-11-07 19:04:53 +01:00
Glenn JocherandGitHub e4f7458d90 Hydra *.yml extension deprecated fix (#34) 2022-11-07 00:52:08 +01:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
6fe8bead35 Classify training cleanup (#33)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-07 00:15:57 +01:00
Glenn JocherandGitHub 7a2e5fdfa3 Update ci.yaml (#21) 2022-09-21 13:57:28 +02:00
Glenn JocherandGitHub abbd177a78 Update README.md 2022-09-21 13:45:12 +02:00
Glenn JocherandGitHub f5dd65c709 Update README.md 2022-09-21 13:42:29 +02:00
Glenn JocherandGitHub 2472feafad Add missing setup.py fields (#18) 2022-09-19 15:49:09 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
171ad46d6b Create __init__.py (#17)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-09-19 15:19:13 +02:00
Glenn JocherandGitHub 64212b67b9 tensorflow-macos comment (#14) 2022-09-18 17:14:46 +02:00
Glenn JocherandGitHub 3c58c26569 Update requirements.txt with fire>=0.4.0 (#12) 2022-09-17 11:41:08 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
088bd5b462 Create ci.yaml (#10)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-09-17 00:50:20 +02:00
Glenn JocherandGitHub 4aa74bb726 Normalize version with dev0 (#11) 2022-09-17 00:06:11 +02:00
Glenn JocherandGitHub 745f8e3759 Update requirements.txt (#9) 2022-09-16 23:42:08 +02:00
Glenn JocherandGitHub 95f6fa16d5 Initial commit 2022-09-11 19:39:46 +03:00