Glenn Jocher
8e60fc9276
Add `device=cuda` support ( #3133 )
1 year 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>
1 year ago
Glenn Jocher
e70de6dacb
`ultralytics 8.0.115` RayTune and DVC fixes ( #3115 )
1 year ago
Ivan Shcheklein
3892e38021
fix(loggers): DVCLive add package version check ( #3112 )
1 year ago
Siddhant Sadangi
29c954a138
Remove Neptune `run.stop()` in `on_train_end()` ( #3103 )
1 year ago
Glenn Jocher
e88c89d269
Fix RayTune for < 8 CPUs ( #3102 )
1 year 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>
1 year ago
Glenn Jocher
8ac8ff72ae
Fix code scanning alert - Incomplete URL substring sanitization ( #3054 )
1 year ago
Laughing
31281400dd
Simplify `remove_zero_areas()` function ( #3060 )
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>
1 year 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>
1 year ago
Ivan Shcheklein
6057b267af
Add DVC experiments logger with DVCLive ( #2792 )
1 year ago
Glenn Jocher
2b26572e42
Fix GlobalWheat2020.yaml auto download ( #3007 )
1 year ago
Glenn Jocher
2193ca1658
`ultralytics 8.0.113` Adamax, NAdam, RAdam optimizers ( #3004 )
1 year 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>
1 year ago
Laughing
f502b50365
fix overlap_mask in SegmentationLoss ( #2985 )
1 year ago
Laughing
bb5bbc2893
Fix keypoints.conf update Results docs ( #2977 )
1 year 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>
1 year 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>
1 year ago
Glenn Jocher
d07ba25dc4
`ultralytics 8.0.112` augment, results and docs updates ( #2944 )
1 year 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>
1 year ago
Glenn Jocher
96fb5ee78e
Update license and tasks docs ( #2941 )
1 year ago
Laughing
c050b2d1a8
Task augment ( #2924 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 year 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>
1 year ago
Ayush Chaurasia
6c65934b55
W&B updates ( #2895 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 year ago
Glenn Jocher
90ec434ebb
Update stream_loaders.py to `Path().absolute()` ( #2915 )
1 year 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>
1 year ago
Laughing
305cde69d0
Update `generate_ddp_file` for improved `overrides` ( #2909 )
...
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
1 year 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>
1 year 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>
1 year 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>
1 year ago
Glenn Jocher
0d47d11393
Fix DDP classify train `imgsz` ( #2897 )
1 year ago
Glenn Jocher
a2bb42dfe9
Fix `Conv2()` fusing ( #2885 )
1 year ago
Glenn Jocher
5baaf91b5b
Add ProfileModel `min_time=60`s argument ( #2881 )
1 year ago
Glenn Jocher
6bf5000a88
`ultralytics 8.0.110` new `profile` and `fraction` train args ( #2880 )
1 year 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>
1 year ago
Glenn Jocher
dfe7f627d8
Remove Sentry from requirements ( #2878 )
1 year ago
John-Henry Lim
1a26e1e195
kwargs override `cfg` argument in model.train ( #2839 ) ( #2853 )
...
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
1 year ago
setwgx
0c62f85750
Fix KeyError:line_thickness ( #2852 )
...
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
1 year ago
Laughing
ef1eff8830
Fix numpy FP64 bug in augmentation ( #2841 )
...
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
1 year 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>
1 year 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>
1 year ago
Glenn Jocher
0bdd4ad379
New dataset `fraction=1.0` argument ( #2860 )
1 year 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>
1 year ago
Glenn Jocher
441e67d330
Add Conv2() module ( #2820 )
1 year 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>
1 year 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>
1 year 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>
1 year 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>
1 year 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