234 Commits

Author SHA1 Message Date
Soumik Rakshit
3821d4a72c
Add alias "best" for the model checkpoint artifact (#3558)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-07-06 14:23:55 +02:00
Glenn Jocher
ad99246ff1
ultralytics 8.0.128 FastSAM autodownload and super() init (#3552)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-07-06 01:11:37 +02:00
Glenn Jocher
4c22273198
ultralytics 8.0.126 Ray Tune refactoring (#3511)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-07-05 00:30:17 +02:00
Glenn Jocher
8a11eda4a9
ultralytics 8.0.125 NMS speed improvements (#3463)
Co-authored-by: Compunet <117437050+dme-compunet@users.noreply.github.com>
2023-07-03 16:34:13 +02:00
CptCaptain
0e5a6b8158
DVCLive cache images (#3459) 2023-07-03 16:29:40 +02:00
Andy
5e38c7d71b
Improve NMS speed (#3467)
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>
2023-07-03 16:20:49 +02:00
Glenn Jocher
05760b04c4
Fix dataset multiple unzip behavior (#3413) 2023-06-27 17:14:39 +02:00
Glenn Jocher
9b00d37022
FROM ubuntu:lunar-20230615 and lap install fixes (#3374) 2023-06-25 21:05:38 +02:00
Glenn Jocher
137552996a
ultralytics 8.0.121 RTDETR fix and Docs updates (#3314) 2023-06-22 00:58:37 +02:00
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>
2023-06-19 22:09:17 +02:00
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>
2023-06-18 16:09:46 +02:00
Glenn Jocher
9d38d2b044
Fix Classify confusion matrix (#3241) 2023-06-18 03:07:45 +02:00
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>
2023-06-17 20:10:47 +02:00
Glenn Jocher
c8441a5422
Fix scipy.ndimage.filters deprecation warning (#3233) 2023-06-17 19:53:41 +02:00
Glenn Jocher
bd0f7ecf6f
Docstrings arguments cleanup (#3229)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-06-17 16:17:07 +02:00
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>
2023-06-17 13:46:18 +02:00
Glenn Jocher
03bce07848
Build reference for missing __init__.py files (#3227) 2023-06-17 13:31:46 +02:00
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>
2023-06-17 01:07:24 +02:00
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>
2023-06-16 02:30:58 +02:00
Dhruv Nair
a0fa99c4db
Comet Updates (#3186)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-06-16 02:06:28 +02:00
Glenn Jocher
f8e1dcc43f
Fix operating system to list (#3196) 2023-06-16 02:04:31 +02:00
Laughing
b59342b81c
Fix platform set to tuple (#3136)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-06-12 10:35:37 +02:00
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>
2023-06-11 20:39:32 +02:00
Ivan Shcheklein
9e859c949e
Use model_info_for_loggers() in DVC logger (#3127) 2023-06-11 17:59:42 +02:00
Glenn Jocher
8e60fc9276
Add device=cuda support (#3133) 2023-06-11 17:58:10 +02:00
Ivan Shcheklein
3892e38021
fix(loggers): DVCLive add package version check (#3112) 2023-06-09 11:19:34 +02:00
Siddhant Sadangi
29c954a138
Remove Neptune run.stop() in on_train_end() (#3103) 2023-06-08 19:16:51 +02:00
Glenn Jocher
e88c89d269
Fix RayTune for < 8 CPUs (#3102) 2023-06-08 18:50:05 +02:00
Laughing
31281400dd
Simplify remove_zero_areas() function (#3060) 2023-06-07 08:13:56 +02:00
Ivan Shcheklein
6057b267af
Add DVC experiments logger with DVCLive (#2792) 2023-06-05 17:10:43 +02:00
Glenn Jocher
2b26572e42
Fix GlobalWheat2020.yaml auto download (#3007) 2023-06-05 01:37:36 +02:00
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>
2023-06-04 22:35:50 +02:00
Laughing
f502b50365
fix overlap_mask in SegmentationLoss (#2985) 2023-06-04 22:35:33 +02:00
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>
2023-06-02 15:43:07 +02:00
Ayush Chaurasia
6c65934b55
W&B updates (#2895)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-06-01 00:33:06 +02:00
Laughing
305cde69d0
Update generate_ddp_file for improved overrides (#2909)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-05-30 19:41:44 +02:00
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>
2023-05-30 19:24:30 +02:00
Glenn Jocher
5baaf91b5b
Add ProfileModel min_time=60s argument (#2881) 2023-05-28 20:07:17 +02:00
Glenn Jocher
dfe7f627d8
Remove Sentry from requirements (#2878) 2023-05-28 17:40:59 +02:00
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>
2023-05-28 16:15:41 +02:00
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>
2023-05-28 03:51:49 +02:00
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>
2023-05-25 16:07:54 +05:30
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>
2023-05-25 00:43:32 +02:00
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>
2023-05-25 00:42:13 +02:00
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>
2023-05-23 13:50:24 +02:00
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>
2023-05-22 13:31:19 +02:00
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>
2023-05-17 19:10:20 +02:00
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>
2023-05-17 02:26:01 +02:00
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>
2023-05-16 01:39:52 +02:00
Laughing
ff211f4037
fix loss (#2614) 2023-05-15 12:09:34 +02:00