Commit Graph

459 Commits

Author SHA1 Message Date
cd0bf05aa4 Cap head scaling with min(nc, 100) (#3436) 2023-06-28 14:11:35 +02:00
17e139220b ultralytics 8.0.124 lapx package and multiple-unzip fix (#3423) 2023-06-28 00:36:01 +02:00
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
05760b04c4 Fix dataset multiple unzip behavior (#3413) 2023-06-27 17:14:39 +02:00
42057fdc95 lapx package replacement for lap (#3383) 2023-06-26 21:30:14 +02:00
e93a5fbff6 Fix RTDETR val_batch_pred (#3392) 2023-06-26 13:37:33 +02:00
b5a2be8ca8 ultralytics 8.0.123 Ubuntu security and VideoWriter codec fixes (#3380) 2023-06-25 23:59:28 +02:00
8b50b2a897 Fix video writing (#3379) 2023-06-25 23:11:00 +02:00
9b00d37022 FROM ubuntu:lunar-20230615 and lap install fixes (#3374) 2023-06-25 21:05:38 +02:00
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
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
51d8cfa9c3 Fix RT-DETR exported onnx model (#3317)
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>
2023-06-24 16:34:24 +02:00
5381fc8a58 Warn save disabled for torch inference (#3361) 2023-06-24 15:57:17 +02:00
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
137552996a ultralytics 8.0.121 RTDETR fix and Docs updates (#3314) 2023-06-22 00:58:37 +02:00
35c02c8319 Fix Pose export from YAML (#3312) 2023-06-21 18:04:22 +02:00
9d1e5567de [RTDETR]Fix val loss (#3280) 2023-06-20 17:59:04 +02:00
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
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
eda7758521 [RTDETR]fix "Grad strides do not match bucket view strides" when training with DDP (#3255) 2023-06-19 16:09:06 +02:00
58bccb1a9f Add CLI support for SAM, RTDETR (#3253) 2023-06-18 22:47:33 +02:00
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
064d48eff9 ultralytics 8.0.119 RTDETR Trainer and ViT Docs (#3236)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-06-18 03:52:08 +02:00
bba7cf55a0 Fix data.yaml absolute path bug (#3242) 2023-06-18 03:35:24 +02:00
9d38d2b044 Fix Classify confusion matrix (#3241) 2023-06-18 03:07:45 +02:00
9c44560ec4 Return save_dir with prediction results (#3240) 2023-06-18 02:34:31 +02:00
15c90bd404 Default classify training to pretrained=True (#3239) 2023-06-18 02:31:05 +02:00
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
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
c8441a5422 Fix scipy.ndimage.filters deprecation warning (#3233) 2023-06-17 19:53:41 +02:00
d3ca3e4b9b Fix Pose val plotting bug (#3232) 2023-06-17 19:35:19 +02:00
1e5702a5b5 Add Ultralytics ViT Docs (#3230)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-06-17 16:44:41 +02:00
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
62916b3b0a docs: add docs for hub datasets (#3212)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-06-17 14:05:41 +02:00
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
03bce07848 Build reference for missing __init__.py files (#3227) 2023-06-17 13:31:46 +02:00
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
d69a1e8046 Update save_dir on new predict args (#3215) 2023-06-16 19:59:46 +02:00
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
a0fa99c4db Comet Updates (#3186)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-06-16 02:06:28 +02:00
0d91d6df6e Add world_size check before setting up DDP train (#3191) 2023-06-16 02:05:15 +02:00
f8e1dcc43f Fix operating system to list (#3196) 2023-06-16 02:04:31 +02:00
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
b59342b81c Fix platform set to tuple (#3136)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-06-12 10:35:37 +02:00
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
9e859c949e Use model_info_for_loggers() in DVC logger (#3127) 2023-06-11 17:59:42 +02:00
8e60fc9276 Add device=cuda support (#3133) 2023-06-11 17:58:10 +02:00
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
e70de6dacb ultralytics 8.0.115 RayTune and DVC fixes (#3115) 2023-06-09 11:21:03 +02:00
3892e38021 fix(loggers): DVCLive add package version check (#3112) 2023-06-09 11:19:34 +02:00