Glenn Jocher
8b50b2a897
Fix video writing ( #3379 )
2 years ago
Glenn Jocher
9b00d37022
FROM ubuntu:lunar-20230615 and `lap` install fixes ( #3374 )
2 years ago
Glenn Jocher
15c027a9fc
Add `torch.Tensor` checks and pip badges ( #3368 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years 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>
2 years ago
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>
2 years ago
Glenn Jocher
5381fc8a58
Warn `save` disabled for torch inference ( #3361 )
2 years 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>
2 years ago
Glenn Jocher
137552996a
`ultralytics 8.0.121` RTDETR fix and Docs updates ( #3314 )
2 years ago
Glenn Jocher
35c02c8319
Fix Pose export from YAML ( #3312 )
2 years ago
Laughing
9d1e5567de
[RTDETR]Fix val loss ( #3280 )
2 years ago
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>
2 years ago
Jeroen
2e8313b032
Allow setting MLFlow experiment name through env var ( #3266 )
...
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
eda7758521
[RTDETR]fix "Grad strides do not match bucket view strides" when training with DDP ( #3255 )
2 years ago
Glenn Jocher
58bccb1a9f
Add CLI support for SAM, RTDETR ( #3253 )
2 years ago
Glenn Jocher
4c2033d7c3
Uninstall `wandb` from notebook environments ( #3247 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
064d48eff9
`ultralytics 8.0.119` RTDETR Trainer and ViT Docs ( #3236 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
bba7cf55a0
Fix data.yaml absolute path bug ( #3242 )
2 years ago
Glenn Jocher
9d38d2b044
Fix Classify confusion matrix ( #3241 )
2 years ago
Glenn Jocher
9c44560ec4
Return `save_dir` with prediction results ( #3240 )
2 years ago
Glenn Jocher
15c90bd404
Default classify training to `pretrained=True` ( #3239 )
2 years ago
Glenn Jocher
e78fb683f4
`default.yaml` type comments ( #3237 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
21df296425
Fix PIL `getsize()` deprecation warning ( #3234 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
c8441a5422
Fix `scipy.ndimage.filters` deprecation warning ( #3233 )
2 years ago
Glenn Jocher
d3ca3e4b9b
Fix Pose val plotting bug ( #3232 )
2 years ago
Glenn Jocher
1e5702a5b5
Add Ultralytics ViT Docs ( #3230 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
bd0f7ecf6f
Docstrings arguments cleanup ( #3229 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Sergiu Waxmann
62916b3b0a
docs: add docs for hub datasets ( #3212 )
...
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years 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>
2 years ago
Glenn Jocher
03bce07848
Build reference for missing `__init__.py` files ( #3227 )
2 years 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>
2 years ago
Glenn Jocher
d69a1e8046
Update save_dir on new predict args ( #3215 )
2 years ago
Glenn Jocher
a5a34af2f4
`ultralytics 8.0.118` updated Docs, OS and DDP fixes ( #3197 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Dhruv Nair
a0fa99c4db
Comet Updates ( #3186 )
...
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
Bruno Arine
0d91d6df6e
Add world_size check before setting up DDP train ( #3191 )
2 years ago
Glenn Jocher
f8e1dcc43f
Fix operating system to list ( #3196 )
2 years 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>
2 years ago
Laughing
b59342b81c
Fix platform set to tuple ( #3136 )
...
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
Glenn Jocher
f4e8b39fc2
`ultralytics 8.0.116` NAS, DVC, YOLOv5u updates ( #3124 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Ivan Shcheklein
9e859c949e
Use `model_info_for_loggers()` in DVC logger ( #3127 )
2 years ago
Glenn Jocher
8e60fc9276
Add `device=cuda` support ( #3133 )
2 years ago
Glenn Jocher
ff91fbd9c3
Add Ultralytics tasks and YOLO-NAS models ( #2735 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
e70de6dacb
`ultralytics 8.0.115` RayTune and DVC fixes ( #3115 )
2 years ago
Ivan Shcheklein
3892e38021
fix(loggers): DVCLive add package version check ( #3112 )
2 years ago
Siddhant Sadangi
29c954a138
Remove Neptune `run.stop()` in `on_train_end()` ( #3103 )
2 years ago
Glenn Jocher
e88c89d269
Fix RayTune for < 8 CPUs ( #3102 )
2 years ago
Glenn Jocher
3ae81ee9d1
Update val `max_dets=args.max_det=300` ( #3051 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
8ac8ff72ae
Fix code scanning alert - Incomplete URL substring sanitization ( #3054 )
2 years ago
Laughing
31281400dd
Simplify `remove_zero_areas()` function ( #3060 )
2 years 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
Ivan Shcheklein
6057b267af
Add DVC experiments logger with DVCLive ( #2792 )
2 years ago
Glenn Jocher
2b26572e42
Fix GlobalWheat2020.yaml auto download ( #3007 )
2 years ago
Glenn Jocher
2193ca1658
`ultralytics 8.0.113` Adamax, NAdam, RAdam optimizers ( #3004 )
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
f502b50365
fix overlap_mask in SegmentationLoss ( #2985 )
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
Glenn Jocher
d07ba25dc4
`ultralytics 8.0.112` augment, results and docs updates ( #2944 )
2 years ago
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>
2 years ago
Glenn Jocher
96fb5ee78e
Update license and tasks docs ( #2941 )
2 years ago
Laughing
c050b2d1a8
Task augment ( #2924 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.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
Ayush Chaurasia
6c65934b55
W&B updates ( #2895 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
90ec434ebb
Update stream_loaders.py to `Path().absolute()` ( #2915 )
2 years ago
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>
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
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>
2 years ago
tekert
50c9731808
Refactor best YouTube URL and add `yt_dlp` option ( #2901 )
...
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
0d47d11393
Fix DDP classify train `imgsz` ( #2897 )
2 years ago
Glenn Jocher
a2bb42dfe9
Fix `Conv2()` fusing ( #2885 )
2 years ago
Glenn Jocher
5baaf91b5b
Add ProfileModel `min_time=60`s argument ( #2881 )
2 years ago
Glenn Jocher
6bf5000a88
`ultralytics 8.0.110` new `profile` and `fraction` train args ( #2880 )
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
Glenn Jocher
dfe7f627d8
Remove Sentry from requirements ( #2878 )
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
Laughing
ef1eff8830
Fix numpy FP64 bug in augmentation ( #2841 )
...
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
6391c60089
New train `profile` argument for loggers ( #2862 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
0bdd4ad379
New dataset `fraction=1.0` argument ( #2860 )
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
Glenn Jocher
441e67d330
Add Conv2() module ( #2820 )
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
f23a03596d
`ultralytics 8.0.109` HUB training fix ( #2818 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.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
07b57c03c8
Buffered Mosaic for reduced HDD reads ( #2791 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.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
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>
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
Laughing
ff211f4037
fix loss ( #2614 )
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
d1107ca4cb
`ultralytics 8.0.97` confusion matrix, windows, docs updates ( #2511 )
...
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: Dowon <ks2515@naver.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
6ee3a9a74b
`ultralytics 8.0.96` TAL speed and memory improvements ( #2484 )
...
Signed-off-by: Evangelos Petrongonas <e.petrongonas@hellenicdrones.com>
Co-authored-by: Evangelos Petrongonas <24351757+vpetrog@users.noreply.github.com>
Co-authored-by: JF Chen <k-2feng@hotmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago