Glenn Jocher
15c90bd404
Default classify training to `pretrained=True` ( #3239 )
1 year 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
Glenn Jocher
0d47d11393
Fix DDP classify train `imgsz` ( #2897 )
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
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
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
e21428ca4e
`ultralytics 8.0.94` HUBDatasetStats() Segment and Pose support ( #2450 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: JF Chen <k-2feng@hotmail.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
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
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
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
Glenn Jocher
31db8ed163
`ultralytics 8.0.79` expand Docs reference section ( #2053 )
...
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: Fri3dChicken <87434761+AmoghDhaliwal@users.noreply.github.com>
2 years ago
Glenn Jocher
2c6fc0a444
Update LICENSE to AGPL-3.0 ( #2031 )
2 years ago
Glenn Jocher
2725545090
`ultralytics 8.0.67` Pose speeds, Comet and ClearML updates ( #1871 )
...
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: Victor Sonck <victor.sonck@gmail.com>
Co-authored-by: Danny Kim <dh031200@gmail.com>
2 years ago
Glenn Jocher
3ea659411b
`ultralytics 8.0.44` export and task fixes ( #1088 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Mehran Ghandehari <mehran.maps@gmail.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
2 years ago
Glenn Jocher
edd3ff1669
Update `.pre-commit-config.yaml` ( #1026 )
2 years ago
Glenn Jocher
1ad7e79033
Fix `yolo checks` as a package bug in Colab ( #972 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Sergio Sanchez <sergio.ssm.97@gmail.com>
2 years ago
Glenn Jocher
d32b339373
`ultralytics 8.0.34` security updates and fixes ( #924 )
...
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: Fabian Greavu <fabiangreavu@gmail.com>
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
7a7c8dc7b7
`ultralytics 8.0.29` DDP-cls and default arg fixes ( #813 )
2 years ago
Glenn Jocher
899abe9f82
`imgsz` warning fix, download function consolidation ( #681 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: HaeJin Lee <seareale@gmail.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
2 years ago
Glenn Jocher
a5410ed79e
DDP, Comet, URLError fixes, improved error handling ( #658 )
...
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: Tungway1990 <68179274+Tungway1990@users.noreply.github.com>
2 years ago
Glenn Jocher
15b3b0365a
`ultralytics 8.0.20` CLI `yolo` simplifications, DDP and ONNX fixes ( #608 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Sid Prabhakaran <s2siddhu@gmail.com>
2 years ago
Glenn Jocher
520825c4b2
`ultralytics 8.0.19` seg/det dataset warning and DDP-cls/seg fixes ( #595 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: 曾逸夫(Zeng Yifu) <41098760+Zengyf-CVer@users.noreply.github.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
2 years ago
Glenn Jocher
c5fccc3fc4
`ultralytics 8.0.12` - Hydra removal ( #506 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Pronoy Mandal <lukex9442@gmail.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
2 years ago
Glenn Jocher
a86218b767
Fix model re-fuse() in inference loops ( #466 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
2 years ago
Ayush Chaurasia
c6985da9de
New YOLOv8 `Results()` class for prediction outputs ( #314 )
...
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: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: Viet Nhat Thai <60825385+vietnhatthai@users.noreply.github.com>
Co-authored-by: Paula Derrenger <107626595+pderrenger@users.noreply.github.com>
2 years ago
Glenn Jocher
583eac0e80
General refactoring and improvements ( #373 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
f8e32c4c13
General `ultralytics==8.0.6` updates ( #351 )
...
Co-authored-by: Dzmitry Plashchynski <plashchynski@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Ayush Chaurasia
216cf2ddb6
Release 8.0.4 fixes ( #256 )
...
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 <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: TechieG <35962141+gokulnath30@users.noreply.github.com>
Co-authored-by: Parthiban Marimuthu <66585214+partheee@users.noreply.github.com>
2 years ago
Glenn Jocher
c7629e93bd
Add YOLOv5 dataset yamls ( #207 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
f80ff923e7
Start Multi-OS CI ( #172 )
2 years ago
Glenn Jocher
422c49d439
Cleanup ( #168 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Kalen Michael <kalenmike@gmail.com>
2 years ago
Ayush Chaurasia
cb4801888e
README and Docs updates ( #166 )
...
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
e79ea1666c
Fix Classification train logging ( #157 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
2 years ago
Ayush Chaurasia
07eab49c3d
Add Classification model YAML support ( #154 )
...
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
Ayush Chaurasia
0e5a7ae623
General trainer cleanup ( #147 )
...
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>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
2 years ago
Glenn Jocher
92dad1c1b5
Start export implementation ( #110 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Ayush Chaurasia
100e21ee80
fix multiple dirs in DDP mode ( #113 )
...
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
6f0ba81427
Add best.pt val and COCO pycocotools val ( #98 )
...
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>
2 years ago
Glenn Jocher
48cffa176e
Shorten module paths with new 'nn' dir ( #96 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
98815d560f
YOLOv5 updates ( #90 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
6432afc5f9
Rename `img_size` to `imgsz` ( #86 )
2 years ago
Ayush Chaurasia
7ae45c6cc4
[WIP] Model interface ( #68 )
...
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
Laughing
fbeeb5d1e1
add resuming ( #63 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Ayush Chaurasia
5a52e7663a
standalone val ( #56 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Laughing
3a241e4cea
update segment training ( #57 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: ayush chaurasia <ayush.chaurarsia@gmail.com>
2 years ago
Ayush Chaurasia
832ea56eb4
update model initialization design, supports custom data/num_classes ( #44 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Ayush Chaurasia
1f3aad86c1
new check_dataset functions ( #43 )
...
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago