344 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
Wenchao Ding
400f3f72a1
ultralytics 8.0.127 add FastSAM model (#3390)
Co-authored-by: dingwenchao <12962189468@163.com>
Co-authored-by: 丁文超 <dingwenchao@dingwenchaodeMacBook-Pro.local>
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>
2023-07-06 00:16:22 +02:00
Glenn Jocher
91905b4b0b
Pass export=True to RTDETRDecoder (#3550)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-07-05 18:25:27 +02:00
Glenn Jocher
74275655dc
Update default.yaml imgsz comment (#3551) 2023-07-05 17:55:04 +02:00
Glenn Jocher
ee6cf11378
Save Keypoints visible to text (#3547)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-07-05 17:14:53 +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
25159c9376
Set close_mosaic: 10 in default.yaml (#3525) 2023-07-04 13:17:31 +02:00
Laughing
8c0a7440df
[RTDETR]fix mosaic<1.0 (#3518) 2023-07-04 10:09: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
Neloy262
586c95b8a9
Add torch.Tensor results image saving (#3475)
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:08:02 +02:00
Liam Parker
27a24b2796
Change key format for images dict in convert_coco (#3419) (#3420)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-06-27 21:01:26 +02:00
Glenn Jocher
05760b04c4
Fix dataset multiple unzip behavior (#3413) 2023-06-27 17:14:39 +02:00
Glenn Jocher
8b50b2a897
Fix video writing (#3379) 2023-06-25 23:11:00 +02:00
Glenn Jocher
9b00d37022
FROM ubuntu:lunar-20230615 and lap install fixes (#3374) 2023-06-25 21:05:38 +02:00
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>
2023-06-25 02:52:00 +02:00
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>
2023-06-25 01:36:07 +02:00
Glenn Jocher
5381fc8a58
Warn save disabled for torch inference (#3361) 2023-06-24 15:57:17 +02:00
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>
2023-06-24 15:56:18 +02:00
Glenn Jocher
137552996a
ultralytics 8.0.121 RTDETR fix and Docs updates (#3314) 2023-06-22 00:58:37 +02:00
Glenn Jocher
35c02c8319
Fix Pose export from YAML (#3312) 2023-06-21 18:04:22 +02:00
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>
2023-06-19 22:12:48 +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
58bccb1a9f
Add CLI support for SAM, RTDETR (#3253) 2023-06-18 22:47:33 +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
bba7cf55a0
Fix data.yaml absolute path bug (#3242) 2023-06-18 03:35:24 +02:00
Glenn Jocher
9d38d2b044
Fix Classify confusion matrix (#3241) 2023-06-18 03:07:45 +02:00
Glenn Jocher
9c44560ec4
Return save_dir with prediction results (#3240) 2023-06-18 02:34:31 +02:00
Glenn Jocher
15c90bd404
Default classify training to pretrained=True (#3239) 2023-06-18 02:31:05 +02:00
Glenn Jocher
e78fb683f4
default.yaml type comments (#3237)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-06-18 01:24:59 +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
d3ca3e4b9b
Fix Pose val plotting bug (#3232) 2023-06-17 19:35:19 +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
d69a1e8046
Update save_dir on new predict args (#3215) 2023-06-16 19:59:46 +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
Bruno Arine
0d91d6df6e
Add world_size check before setting up DDP train (#3191) 2023-06-16 02:05:15 +02:00
Glenn Jocher
f8e1dcc43f
Fix operating system to list (#3196) 2023-06-16 02:04:31 +02:00
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>
2023-06-12 20:14:46 +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
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>
2023-06-10 18:54:14 +02:00