50 Commits

Author SHA1 Message Date
Glenn Jocher
edd3ff1669
Update .pre-commit-config.yaml (#1026) 2023-02-17 22:26:40 +01:00
Glenn Jocher
977fd8f0b8
ultralytics 8.0.35 TensorRT, ONNX and OpenVINO predict and val (#929)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Eric Pedley <ericpedley@gmail.com>
2023-02-11 21:31:49 +04:00
Glenn Jocher
254adfa652
ultralytics 8.0.33 security updates and fixes (#896)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Mert Can Demir <validatedev@gmail.com>
2023-02-10 01:37:57 +04:00
Glenn Jocher
c9893810c7
ultralytics 8.0.32 HUB and TensorFlow fixes (#870)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-02-09 01:47:34 +04:00
Glenn Jocher
64f247d692
ultralytics 8.0.30 Docker, rect, data=*.zip updates (#832)
Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
2023-02-06 21:57:10 +04:00
Glenn Jocher
6c44ce21d9
ultralytics 8.0.21 Windows, segments, YAML fixes (#655)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: corey-nm <109536191+corey-nm@users.noreply.github.com>
2023-01-26 23:07:27 +01:00
Glenn Jocher
520825c4b2
ultralytics 8.0.19 seg/det dataset warning and DDP-cls/seg fixes (#595)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: 曾逸夫(Zeng Yifu) <41098760+Zengyf-CVer@users.noreply.github.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
2023-01-24 23:22:02 +01:00
Ayush Chaurasia
936414c615
ultralytics 8.0.18 new python callbacks and minor fixes (#580)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Jeroen Rombouts <36196499+jarombouts@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-01-23 18:31:29 +01:00
Glenn Jocher
d9a0fba251
ultralytics 8.0.14 Hydra removal fixes and cleanup (#542)
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>
Co-authored-by: Kamlesh Kumar <patelkamleshpatel364@gmail.com>
2023-01-21 21:22:40 +01:00
Glenn Jocher
c5fccc3fc4
ultralytics 8.0.12 - Hydra removal (#506)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Pronoy Mandal <lukex9442@gmail.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
2023-01-21 00:45:19 +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
314da263c7
Default confidences predict 0.25, val 0.001 (#149) 2023-01-05 15:56:52 +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
Glenn Jocher
3c18aa5278
Simplify argument names (#141) 2023-01-03 19:47:56 +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
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
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
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
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
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
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
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>
2022-12-27 15:49:13 +01:00
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>
2022-12-27 04:56:24 +01:00
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>
2022-12-26 17:34:28 +01:00
Glenn Jocher
249dfbdc05
Fix ClearML Mosaic callback to 'on_train_epoch_end' (#92) 2022-12-25 17:04:24 +01:00
Glenn Jocher
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 Jocher
6432afc5f9
Rename img_size to imgsz (#86) 2022-12-24 00:39:09 +01:00
Ayush Chaurasia
ae2443c210
Add flops, num_params, inference speed logging and best.pt logging (#84)
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-23 20:14:21 +01:00
Ayush Chaurasia
e6737f1207
Predictor support (#65)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-07 10:33:10 +05:30
Ayush Chaurasia
5a52e7663a
standalone val (#56)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-30 15:04:44 +05:30
Laughing
3a241e4cea
update segment training (#57)
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-11-29 17:00:08 +05:30
ayush chaurasia
793dde365d Revert "update"
This reverts commit 486bc494142744e9d90f0483b611a785381401a7.
2022-11-25 01:54:50 +05:30
ayush chaurasia
486bc49414 update 2022-11-25 01:11:55 +05:30
Ayush Chaurasia
298287298d
Add clearml logging (#51)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-24 15:25:03 +01:00
Ayush Chaurasia
4291b9c31c
Add EMA and model checkpointing (#49)
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-11-19 19:07:26 +01:00
Glenn Jocher
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
Laughing
47f1cb3ef4
Fix some cuda training issues of segmentation (#46)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-17 18:14:02 +05:30
Ayush Chaurasia
d143ac666f
[rename] - preprocess-batch -> preprocess, preprocess_preds -> postprocess (#42) 2022-11-09 17:35:15 +01:00
Glenn Jocher
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
Ayush Chaurasia
f56c9bcc26
Segmentation support & other enchancements (#40)
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-11-08 16:27:57 +01:00
Ayush Chaurasia
7b560f7861
Model builder (#29)
Co-authored-by: Ayush Chaurasia <ayush.chuararsia@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-18 23:14:23 +02:00
Ayush Chaurasia
c5cb76b356
Metrics and loss structure (#28)
Co-authored-by: Ayush Chaurasia <ayush.chuararsia@gmail.com>
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-10-15 19:39:05 +02:00
Ayush Chaurasia
419f6d7167
Add yolo package structure (#15) 2022-09-19 15:47:23 +05:30