126 Commits

Author SHA1 Message Date
Glenn Jocher
7e99804263
Cleanup argument handling in Model class (#4614)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Sebastian Stapf <42514241+Wiqzard@users.noreply.github.com>
2023-08-28 23:12:32 +02:00
Glenn Jocher
deac7575b1
Model coverage cleanup (#4585) 2023-08-27 04:19:41 +02:00
Glenn Jocher
1db9afc2e5
Fix Windows non-UTF source filenames (#4524)
Co-authored-by: Kayzwer <68285002+Kayzwer@users.noreply.github.com>
2023-08-24 03:22:26 +02:00
Glenn Jocher
6da8f7f51e
Add utils.ops and nn.modules to tests (#4484) 2023-08-22 04:23:28 +02:00
Glenn Jocher
615ddc9d97
Tests and docstrings improvements (#4475) 2023-08-21 17:02:14 +02:00
Glenn Jocher
87ce15d383
ultralytics 8.0.158 add benchmarks to coverage (#4432)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
2023-08-20 20:52:30 +02:00
Glenn Jocher
9d27e7ada4
New ASSETS and trackers GMC cleanup (#4425)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-17 18:19:05 +02:00
Glenn Jocher
fb1ae9bfad
Fix LoadStreams final frame bug (#4387)
Co-authored-by: Nadim Bou Alwan <64587372+nadinator@users.noreply.github.com>
2023-08-16 19:00:03 +02:00
Glenn Jocher
d47718c367
ultralytics 8.0.154 add freeze training argument (#4329) 2023-08-14 00:18:32 +02:00
Glenn Jocher
9f6d48d3cf
Improve tests coverage and speed (#4340)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-13 22:24:01 +02:00
Glenn Jocher
28893d9d2d
ultralytics 8.0.152 bump torch>=1.8.0 (#4299) 2023-08-11 00:49:55 +02:00
Glenn Jocher
7dfdb63cde
ultralytics 8.0.150 new Apple *.mlpackage export format (#4043)
Co-authored-by: MLBoy_DaisukeMajima <rockyshikoku@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Louis Lac <lac.louis5@gmail.com>
2023-08-08 17:00:42 +02:00
Glenn Jocher
7565210484
ultralytics 8.0.149 add Open Images V7 training (#4178)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: AdiEcho <30563671+AdiEcho@users.noreply.github.com>
2023-08-07 01:19:51 +02:00
Glenn Jocher
11d0488bf1
Update prediction Results docs (#4139)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-03 18:44:40 +02:00
Sebastian Stapf
8870084645
Removed redundency in RTDETRDecoder (#4118)
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>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
2023-08-03 18:14:37 +02:00
Glenn Jocher
379f31904a
Fix per-layer FLOPs profiling (#4048)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-07-30 22:24:28 +02:00
Motoki Kimura
3c787eb080
Fix catastrophic accuracy degradation of TFLite static quantized integer models (#1695)
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-07-26 19:12:49 +02:00
Birk Stoveland
42afe772d5
Update tasks.py to allow custom RTEDTRDecoder arguments (#3872)
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-07-23 15:09:12 +02:00
nihui
01dcd54b19
Add ncnn auto thread count (#3847)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-07-21 01:38:59 +02:00
Colin Wong
b251b7415a
Fix tflite int8 scaling (#3837)
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-07-21 01:37:41 +02:00
nihui
cf3c740fe7
update cli.md for export ncnn model (#3846) 2023-07-21 01:19:39 +02:00
Laughing
620f3eb218
ultralytics 8.0.136 refactor and simplify package (#3748)
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-07-16 17:47:45 +02:00
Glenn Jocher
114470361e
Remove deprecated AutoShape class (#3743) 2023-07-15 01:00:23 +02:00
Glenn Jocher
135a10f1fa
Avoid CUDA round-trip for relevant export formats (#3727)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-07-14 20:38:31 +02:00
Harsh Dhamecha
c5991d7cd8
Updated C2f docstring (#3732)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-07-14 14:46:03 +02:00
Adrian Boguszewski
e2f9bba964
Update OpenVINO requirement to openvino>=2023.0 (#3722)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-07-13 18:13:55 +02:00
Glenn Jocher
c55a98ab8e
ultralytics 8.0.133 add torchvision compatibility check (#3703)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-07-13 01:40:50 +02:00
Adrian Boguszewski
0821ccb618
Changed default device to AUTO in OpenVINO (#3699)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-07-12 20:48:11 +02:00
triple Mu
82920ef7ec
Replace an easy way to construct a ncnn.Mat (#3638) 2023-07-11 16:09:32 +02:00
Glenn Jocher
495edc261f
ultralytics 8.0.132 add AutoBackend NCNN inference (#3615)
Co-authored-by: triple Mu <gpu@163.com>
2023-07-10 16:49:51 +02:00
Colin Wong
23c7cd4c9f
Avoid Python 3.9 with_stem (#3589)
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-07-08 00:16:42 +02:00
Glenn Jocher
abb0939fe8
ultralytics 8.0.129 add YOLOv8 Tencent NCNN export (#3529)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-07-07 15:17:38 +02:00
Glenn Jocher
8a11eda4a9
ultralytics 8.0.125 NMS speed improvements (#3463)
Co-authored-by: Compunet <117437050+dme-compunet@users.noreply.github.com>
2023-07-03 16:34:13 +02:00
Glenn Jocher
cd0bf05aa4
Cap head scaling with min(nc, 100) (#3436) 2023-06-28 14:11:35 +02:00
Kayzwer
51d8cfa9c3
Fix RT-DETR exported onnx model (#3317)
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-06-24 16:34:24 +02:00
Laughing
9d1e5567de
[RTDETR]Fix val loss (#3280) 2023-06-20 17:59:04 +02:00
Glenn Jocher
d8701b42ca
ultralytics 8.0.120 CLI support for SAM, RTDETR (#3273)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-06-19 22:12:48 +02:00
Laughing
eda7758521
[RTDETR]fix "Grad strides do not match bucket view strides" when training with DDP (#3255) 2023-06-19 16:09:06 +02:00
Glenn Jocher
e78fb683f4
default.yaml type comments (#3237)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-06-18 01:24:59 +02:00
Glenn Jocher
bd0f7ecf6f
Docstrings arguments cleanup (#3229)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-06-17 16:17:07 +02:00
Ayush Chaurasia
a0ba8ef5f0
Add RTDETR Trainer (#2745)
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>
Co-authored-by: Kayzwer <68285002+Kayzwer@users.noreply.github.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
2023-06-17 13:46:18 +02:00
Ayush Chaurasia
52ae65ac3b
Optionally train_args on attempt_load_weights() (#2928)
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-06-01 01:02:16 +02:00
Laughing
c050b2d1a8
Task augment (#2924)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-06-01 00:41:10 +02:00
Glenn Jocher
fd94d312da
ultralytics 8.0.111 refactored model.loss() method (#2911)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Snyk bot <snyk-bot@snyk.io>
2023-05-30 20:23:30 +02:00
Ayush Chaurasia
facb7861cf
Move loss calculation to head (#2874)
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>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
2023-05-30 19:24:30 +02:00
Ayush Chaurasia
7f077f7654
Import ultralytics.nn.tasks as ultralytics.nn (#2889)
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-05-30 19:23:39 +02:00
Glenn Jocher
a2bb42dfe9
Fix Conv2() fusing (#2885) 2023-05-29 02:08:45 +02:00
Ayush Chaurasia
527a97759b
Revert loss head PR (#2873)
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-05-28 16:15:41 +02:00
Glenn Jocher
441e67d330
Add Conv2() module (#2820) 2023-05-25 18:12:50 +02:00
Ayush Chaurasia
d19c5b6ce8
Move loss to task heads (#2825)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
2023-05-25 16:07:54 +05:30