Commit graph

376 commits

Author SHA1 Message Date
Mohammed Yasin
e99b778cc4
Fix TFLite error and OpenVINO int8 error at imgsz=32 (#18898)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2025-01-26 14:18:51 +01:00
Mohammed Yasin
de05d1b655
Fix export test matrices to exclude nms from Classify models (#18880)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2025-01-25 21:26:58 +01:00
Mohammed Yasin
9181ff62f5
ultralytics 8.3.67 NMS Export for Detect, Segment, Pose and OBB YOLO models (#18484)
Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
2025-01-24 11:00:36 +01:00
Laughing
d6272f258e
RKNN export clean up (#18841) 2025-01-23 13:28:41 +01:00
Laughing
60d6d011fd
Fix automatic optimizer LR with DDP training (#18842) 2025-01-23 13:26:29 +01:00
Glenn Jocher
5bdcf9116d
Fix missing IMX500 export decorator (#18823)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2025-01-22 13:58:28 +01:00
Laughing
e17c1cdfe0
Fix IMX onnx model path (#18813) 2025-01-22 12:05:38 +01:00
Ivor Zhu
b5e0cee943
ultralytics 8.3.65 Rockchip RKNN Integration for Ultralytics YOLO models (#16308)
Signed-off-by: Francesco Mattioli <Francesco.mttl@gmail.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com>
Co-authored-by: Lakshantha Dissanayake <lakshantha@ultralytics.com>
Co-authored-by: Burhan <Burhan-Q@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
Co-authored-by: Lakshantha Dissanayake <lakshanthad@yahoo.com>
Co-authored-by: Francesco Mattioli <Francesco.mttl@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2025-01-21 02:25:54 +01:00
Mohammed Yasin
a8e2464a9c
Allow tuning directory to be set using name (#18760)
Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2025-01-19 17:33:02 +01:00
Francesco Mattioli
f777b63c8b
ultralytics 8.3.63 IMX500 sudo install fix (#18714)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2025-01-17 12:48:49 +01:00
Mohammed Yasin
d9c374c69b
Fix automatic optimizer LR with DDP training (#18641)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
2025-01-16 22:45:49 +01:00
Glenn Jocher
daaebba220
Simplify thop imports (#18717)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2025-01-16 22:27:27 +01:00
Paula Derrenger
deb87d8ce3
Standardize license headers in Python files (#18696)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2025-01-15 15:36:11 +01:00
Kayzwer
2af30654f0
Remove unused dict values in items() with values() (#18651)
Signed-off-by: Kayzwer <68285002+Kayzwer@users.noreply.github.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2025-01-15 13:52:26 +01:00
Glenn Jocher
34b339d033
Apply ruff==0.9.0 formatting (#18624)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2025-01-10 17:27:22 +01:00
Glenn Jocher
3902e740cf
Apply Ruff 0.9.0 (#18622)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2025-01-10 00:05:09 +01:00
Mohammed Yasin
02e4e984a1
Verify dataset >= batch size on INT8 export calibration (#18611)
Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2025-01-09 10:29:17 +01:00
Glenn Jocher
1f1f355a9e
Add export args to model metadata (#18472)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2025-01-02 21:26:05 +01:00
Jiacong Fang
5d8e15800e
ultralytics 8.3.56 PaddlePaddle GPU Inference support (#18468)
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-12-31 13:57:54 +01:00
Mohammed Yasin
306c5f1d3e
Bump maximum version for onnx2tf to 1.26.3 (#18467)
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
2024-12-31 13:57:12 +01:00
Mohammed Yasin
f932a611ab
Add note that IMX export is only supported for YOLOv8n (#18460)
Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
2024-12-30 23:09:07 +01:00
Mohammed Yasin
0e465175fd
Delete clip_model to avoid errors on export (#18445)
Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-12-30 15:53:07 +01:00
Glenn Jocher
e5e91967d9
PyCharm Code Inspect fixes (#18392)
Signed-off-by: UltralyticsAssistant <web@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-12-25 16:24:29 +01:00
Glenn Jocher
05323919b3
Use Any type-hints for args and kwargs (#18372)
Signed-off-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-12-24 20:32:59 +01:00
Glenn Jocher
0b51e4f489
Add dynamic to approved OpenVINO export args (#18353)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-12-22 18:34:39 +01:00
Mohammed Yasin
1d13575ba1
ultralytics 8.3.53 New Export argument validation (#18185)
Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-12-22 02:02:58 +01:00
Mohammed Yasin
5c2ecb7669
Use shell=True to run hyperparameter tuning (#18284)
Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-12-17 19:27:44 +01:00
Mohammed Yasin
f87b447b2d
Fix model.save() for model YAMLs (#18212)
Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-12-16 11:59:05 +01:00
Mohammed Yasin
8535d36885
Fix untrained warning when training from yaml (#18168)
Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-12-11 21:06:06 +01:00
Laughing
3d52c4fa10
ultralytics 8.3.47 fix softmax and Classify head commonality (#18085)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-12-07 13:55:59 +01:00
Laughing
f6005a9166
Set task for Triton inferenceTriton (#18064) 2024-12-06 11:46:48 +01:00
Mohammed Yasin
a672bf79dd
Set dynamic to True only if imgsz is dynamic for ONNX (#17872)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
2024-12-05 22:24:53 +01:00
Void
29826241a0
__getattr__ support to access YOLO attributes via Model class (#17805)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-11-28 22:51:33 +01:00
Laughing
e1adef7a1d
Fix DLA export (#17765)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Lakshantha Dissanayake <lakshanthad@yahoo.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-11-25 16:46:14 +01:00
Burhan
ee6fde0beb
ultralytics 8.3.37 TensorRT auto-workspace size (#17748)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-11-25 11:33:11 +01:00
Laughing
6a762564c8
Enable model.eval() usage for YOLO class (#17754) 2024-11-25 11:00:24 +01:00
Adrian Boguszewski
426879d80d
ultralytics 8.3.36 unpin OpenVINO ARM install version (#16600)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Lakshantha Dissanayake <lakshanthad@yahoo.com>
Co-authored-by: Lakshantha <lakshantha@ultralytics.com>
Co-authored-by: Francesco Mattioli <Francesco.mttl@gmail.com>
Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-11-22 12:25:42 +01:00
Laughing
b7e9e91d46
ultralytics 8.3.35 enable auto letterbox if model is dynamic (#17687)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-11-21 13:02:48 +01:00
Lakshantha Dissanayake
cb476d2999
Enable MNN on RPi and Jetson (#17583)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-11-18 17:41:15 +01:00
Laughing
4453ddab93
ultralytics 8.3.31 add max_num_obj factor for AutoBatch (#17514)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-11-13 23:51:24 +01:00
Daniel Calvo
ba41def4f9
Fix file_name in save_crop (#17499)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-11-13 20:17:49 +01:00
Laughing
0fa1d7d5a6
ultralytics 8.3.29 Sony IMX500 export (#14878)
Signed-off-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
Co-authored-by: Francesco Mattioli <Francesco.mttl@gmail.com>
Co-authored-by: Lakshantha Dissanayake <lakshantha@ultralytics.com>
Co-authored-by: Lakshantha Dissanayake <lakshanthad@yahoo.com>
Co-authored-by: Chizkiyahu Raful <37312901+Chizkiyahu@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Muhammad Rizwan Munawar <muhammadrizwanmunawar123@gmail.com>
Co-authored-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
2024-11-11 14:20:21 +01:00
Kermit
33cdaa5782
Fix conf overwrite in results.py (#17384)
Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com>
2024-11-08 17:55:44 +01:00
Kilian Schnelle
95368cbe54
Changed model.names return type hint from list to dict[int, str] (#17402) 2024-11-08 16:19:46 +01:00
Paula Derrenger
7373816b23
Fix PyPI downloads links (#17399)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-11-07 17:15:15 +01:00
Laughing
6806f15396
Fix model.end2end assert (#17391)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-11-07 12:20:54 +01:00
Francesco Mattioli
da15e27a4d
Added Error for TFLite int8 end2end model export (#17360)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-11-06 13:34:06 +01:00
Mohammed Yasin
d0abd95f95
Fix error on TensorRT export with float workspace value (#17352) 2024-11-05 10:48:33 +01:00
Glenn Jocher
3a4b65c347
ultralytics 8.3.27 HUB timed training fix (#17298)
Signed-off-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-11-01 01:42:51 +01:00
Lakshantha Dissanayake
daaac35fff
Skip MNN export for Raspberry Pi and NVIDIA Jetson (#17292)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-11-01 01:12:29 +01:00