29 Commits

Author SHA1 Message Date
Awsome
1e1247ddee
Add ResNet50 and ResNet101 backbone RTDETR models (#6661)
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-12-03 16:57:11 +01:00
Glenn Jocher
2e71f7f50e
ultralytics 8.0.221 fix Apple MPS inference bug (#6694)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Johnny <johnnync13@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-11-30 13:46:17 +01:00
Glenn Jocher
0f9f857449
Add docs Ultralytics 文档: - zh/index.md (#5871)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-10-21 20:31:45 +02:00
Glenn Jocher
12e3eef844
ultralytics 8.0.197 save P, R, F1 curves to metrics (#5354)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: erminkev1 <83356055+erminkev1@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Andy <39454881+yermandy@users.noreply.github.com>
2023-10-13 02:49:31 +02:00
Glenn Jocher
7fd5dcbd86
Implement all missing docstrings (#5298)
Co-authored-by: snyk-bot <snyk-bot@snyk.io>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-10-10 20:07:13 +02:00
Glenn Jocher
7517667a33
Add docformatter to pre-commit (#5279)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com>
2023-10-09 02:25:22 +02:00
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
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
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
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
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
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
Harsh Dhamecha
c5991d7cd8
Updated C2f docstring (#3732)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-07-14 14:46:03 +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
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
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
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
Glenn Jocher
a2bb42dfe9
Fix Conv2() fusing (#2885) 2023-05-29 02:08:45 +02:00
Glenn Jocher
441e67d330
Add Conv2() module (#2820) 2023-05-25 18:12:50 +02:00
Glenn Jocher
4db686a315
ultralytics 8.0.106 (#2736)
Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: vyskocj <whiskey1939@seznam.cz>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: triple Mu <gpu@163.com>
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: Laughing <61612323+Laughing-q@users.noreply.github.com>
2023-05-22 13:31:19 +02:00
Glenn Jocher
7884098857
ultralytics 8.0.103 minor fixes (#2634)
Co-authored-by: kssion <appplus@icloud.com>
Co-authored-by: Eli Spizzichino <diramazioni@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: xbkaishui <xxkaishui@gmail.com>
2023-05-16 01:39:52 +02:00
Glenn Jocher
dce4efce48
ultralytics 8.0.100 add Mosaic9() augmentation (#2605)
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: Tommy in Tongji <36354458+TommyZihao@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: BIGBOSS-FOX <47949596+BIGBOSS-FOX@users.noreply.github.com>
Co-authored-by: xbkaishui <xxkaishui@gmail.com>
2023-05-14 20:43:35 +02:00
Glenn Jocher
229119c376
ultralytics 8.0.98 add Baidu RT-DETR models (#2527)
Co-authored-by: Kalen Michael <kalenmike@gmail.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: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: Dowon <ks2515@naver.com>
2023-05-11 01:15:20 +02:00