Neloy262
586c95b8a9
Add torch.Tensor results image saving ( #3475 )
...
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>
1 year ago
Glenn Jocher
8b50b2a897
Fix video writing ( #3379 )
1 year ago
Glenn Jocher
682c9ef70f
`ultralytics 8.0.122` Fix `torch.Tensor` inference ( #3363 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: krzysztof.gonia <4281421+kgonia@users.noreply.github.com>
1 year ago
Glenn Jocher
5381fc8a58
Warn `save` disabled for torch inference ( #3361 )
1 year ago
Jieun Park
2ebd808b69
Increase NCCL timeout from 1 hour to 3 hours ( #3343 )
...
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>
1 year ago
Glenn Jocher
35c02c8319
Fix Pose export from YAML ( #3312 )
1 year ago
Glenn Jocher
9c44560ec4
Return `save_dir` with prediction results ( #3240 )
1 year ago
Glenn Jocher
bd0f7ecf6f
Docstrings arguments cleanup ( #3229 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 year ago
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>
1 year ago
Glenn Jocher
0dfafaf39b
Fix `batch=-1` for exports ( #3217 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 year ago
Glenn Jocher
d69a1e8046
Update save_dir on new predict args ( #3215 )
1 year ago
Bruno Arine
0d91d6df6e
Add world_size check before setting up DDP train ( #3191 )
1 year ago
Glenn Jocher
c340f84ce9
`ultralytics 8.0.117` NAS export, classify and tasks banner URL fixes ( #3145 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 year ago
Glenn Jocher
e88c89d269
Fix RayTune for < 8 CPUs ( #3102 )
1 year ago
Glenn Jocher
8ac8ff72ae
Fix code scanning alert - Incomplete URL substring sanitization ( #3054 )
1 year ago
Glenn Jocher
8940a27bdb
`ultralytics 8.0.114` automatic optimizer selection ( #3037 )
...
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>
2 years ago
Zlobin Vladimir
01273c5ca1
Add rt info for OpenVINO ModelAPI ( #2891 )
...
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: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
2 years ago
Glenn Jocher
451cf8b647
Add Adamax, NAdam, RAdam optimizers ( #2969 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Laughing
bb5bbc2893
Fix keypoints.conf update Results docs ( #2977 )
2 years ago
crbrz
a9129fb40e
Fix `workspace` and `verbose` arguments in TensorRT export ( #2954 )
...
Co-authored-by: crbrz <cristiab@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
Ayush Chaurasia
67cf53b475
Move `check_amp()` to checks.py ( #2948 )
...
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>
2 years ago
Laughing
f4b34fc30b
Fix `save_txt` in track mode and add Keypoints and Probs ( #2921 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Laughing
305cde69d0
Update `generate_ddp_file` for improved `overrides` ( #2909 )
...
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
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>
2 years ago
Afif Al Mamun
75012e575e
Rename predictor postprocess `orig_img` to `orig_imgs` ( #2877 )
...
Signed-off-by: Afif Al Mamun <afifanik@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
John-Henry Lim
1a26e1e195
kwargs override `cfg` argument in model.train ( #2839 ) ( #2853 )
...
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
setwgx
0c62f85750
Fix KeyError:line_thickness ( #2852 )
...
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
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>
2 years ago
Glenn Jocher
61fa5efe6d
`build_optimizer()` assign all parameters ( #2855 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
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>
2 years ago
Glenn Jocher
ffc0e8ccf7
`ultralytics 8.0.108` add Meituan YOLOv6 models ( #2811 )
...
Co-authored-by: Michael Currie <mcurrie@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Hicham Talaoubrid <98521878+HichTala@users.noreply.github.com>
Co-authored-by: Zlobin Vladimir <vladimir.zlobin@intel.com>
Co-authored-by: Szymon Mikler <sjmikler@gmail.com>
2 years ago
Glenn Jocher
dada5b73c4
`ultralytics 8.0.107` ( #2778 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Peter van Lunteren <contact@pvanlunteren.com>
2 years ago
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>
2 years ago
Glenn Jocher
23fc50641c
`ultralytics 8.0.105` classification hyp fix and new `onplot` callbacks ( #2684 )
...
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: Ivan Shcheklein <shcheklein@gmail.com>
2 years ago
Glenn Jocher
b1119d512e
`ultralytics 8.0.104` bug fixes and `thop` dependency removal ( #2665 )
...
Co-authored-by: Kevin Abraham <5976139+abraha2d@users.noreply.github.com>
Co-authored-by: Kevin Abraham <abraha2d@users.noreply.github.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>
2 years ago
Glenn Jocher
e2e3e367a2
`ultralytics 8.0.101` mosaic9() and loss bug fixes ( #2608 )
...
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>
2 years ago
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>
2 years ago
Glenn Jocher
db1c5885d5
`ultralytics 8.0.99` HUB resume fix and Docs updates ( #2567 )
...
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
2 years ago
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>
2 years ago
Glenn Jocher
ddb354ce5e
`ultralytics 8.0.93` HUB docs and JSON2YOLO converter ( #2431 )
...
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: 李际朝 <tubkninght@gmail.com>
Co-authored-by: Danny Kim <imbird0312@gmail.com>
2 years ago
Glenn Jocher
0ebd3f2959
`ultralytics 8.0.92` updates and fixes ( #2361 )
...
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: introvin <vinod.4166@gmail.com>
Co-authored-by: marinmarcillat <58145636+marinmarcillat@users.noreply.github.com>
Co-authored-by: BIGBOSS-FOX <47949596+BIGBOSS-FOX@users.noreply.github.com>
2 years ago
Glenn Jocher
44c7c3514d
`ultralytics 8.0.90` actions and docs improvements ( #2326 )
...
Co-authored-by: calmisential <xinyu_std@163.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: triple Mu <gpu@163.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: ran xiao <ben.xiao@me.com>
Co-authored-by: rxiao <ran.xiao@silverpond.com.au>
2 years ago
Glenn Jocher
243fc4b1fe
`ultralytics 8.0.89` SAM predict and auto-annotate ( #2298 )
...
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: Paula Derrenger <107626595+pderrenger@users.noreply.github.com>
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: Snyk bot <snyk-bot@snyk.io>
Co-authored-by: Laughing-q <1185102784@qq.com>
2 years ago
Glenn Jocher
efc941aa81
`ultralytics 8.0.87` improved Pose models ( #2202 )
...
Signed-off-by: dependabot[bot] <support@github.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>
Co-authored-by: Kirolos Atef <keroatef295@gmail.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
3d60347755
`ultralytics 8.0.86` HUB resume, Classify train and RayTune fixes ( #2200 )
...
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
2 years ago
Glenn Jocher
8f5eeb09f3
`ultralytics 8.0.84` results JSON outputs ( #2171 )
...
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.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: Talia Bender <85292283+taliabender@users.noreply.github.com>
2 years ago
Glenn Jocher
6c082ebd6f
`ultralytics 8.0.83` Neptune AI logging addition ( #2130 )
...
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>
Co-authored-by: Toutatis64 <Toutatis64@users.noreply.github.com>
Co-authored-by: M. Tolga Cangöz <46008593+standardAI@users.noreply.github.com>
Co-authored-by: Talia Bender <85292283+taliabender@users.noreply.github.com>
Co-authored-by: Ophélie Le Mentec <17216799+ouphi@users.noreply.github.com>
Co-authored-by: Kadir Şahin <68073829+ssahinnkadir@users.noreply.github.com>
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
2 years ago
Glenn Jocher
55a03ad85f
`ultralytics 8.0.82` docs updates and fixes ( #2098 )
...
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: Aurelio Losquiño Muñoz <38859113+aurelm95@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Paula Derrenger <107626595+pderrenger@users.noreply.github.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
2 years ago
Glenn Jocher
a38f227672
`ultralytics 8.0.81` single-line docstring updates ( #2061 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
5bce1c3021
`ultralytics 8.0.80` single-line docstring fixes ( #2060 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago