Ivan Shcheklein
|
9e859c949e
|
Use `model_info_for_loggers()` in DVC logger (#3127)
|
1 year ago |
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)
|
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 |
