Commit Graph
398 Commits
Author SHA1 Message Date
KayzwerGitHubGlenn Jocherpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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
Glenn JocherandGitHub 5381fc8a58 Warn save disabled for torch inference (#3361) 2023-06-24 15:57:17 +02:00
Jieun ParkGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
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 JocherandGitHub 137552996a ultralytics 8.0.121 RTDETR fix and Docs updates (#3314) 2023-06-22 00:58:37 +02:00
Glenn JocherandGitHub 35c02c8319 Fix Pose export from YAML (#3312) 2023-06-21 18:04:22 +02:00
LaughingandGitHub 9d1e5567de [RTDETR]Fix val loss (#3280) 2023-06-20 17:59:04 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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
JeroenGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
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
LaughingandGitHub eda7758521 [RTDETR]fix "Grad strides do not match bucket view strides" when training with DDP (#3255) 2023-06-19 16:09:06 +02:00
Glenn JocherandGitHub 58bccb1a9f Add CLI support for SAM, RTDETR (#3253) 2023-06-18 22:47:33 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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 JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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
Glenn JocherandGitHub bba7cf55a0 Fix data.yaml absolute path bug (#3242) 2023-06-18 03:35:24 +02:00
Glenn JocherandGitHub 9d38d2b044 Fix Classify confusion matrix (#3241) 2023-06-18 03:07:45 +02:00
Glenn JocherandGitHub 9c44560ec4 Return save_dir with prediction results (#3240) 2023-06-18 02:34:31 +02:00
Glenn JocherandGitHub 15c90bd404 Default classify training to pretrained=True (#3239) 2023-06-18 02:31:05 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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 JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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 JocherandGitHub c8441a5422 Fix scipy.ndimage.filters deprecation warning (#3233) 2023-06-17 19:53:41 +02:00
Glenn JocherandGitHub d3ca3e4b9b Fix Pose val plotting bug (#3232) 2023-06-17 19:35:19 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn JocherLaughing-qKayzwerLaughing
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 JocherandGitHub 03bce07848 Build reference for missing __init__.py files (#3227) 2023-06-17 13:31:46 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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 JocherandGitHub d69a1e8046 Update save_dir on new predict args (#3215) 2023-06-16 19:59:46 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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
Bruno ArineandGitHub 0d91d6df6e Add world_size check before setting up DDP train (#3191) 2023-06-16 02:05:15 +02:00
Glenn JocherandGitHub f8e1dcc43f Fix operating system to list (#3196) 2023-06-16 02:04:31 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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 ShchekleinandGitHub 9e859c949e Use model_info_for_loggers() in DVC logger (#3127) 2023-06-11 17:59:42 +02:00
Glenn JocherandGitHub 8e60fc9276 Add device=cuda support (#3133) 2023-06-11 17:58:10 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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
Glenn JocherandGitHub e70de6dacb ultralytics 8.0.115 RayTune and DVC fixes (#3115) 2023-06-09 11:21:03 +02:00
Ivan ShchekleinandGitHub 3892e38021 fix(loggers): DVCLive add package version check (#3112) 2023-06-09 11:19:34 +02:00
Siddhant SadangiandGitHub 29c954a138 Remove Neptune run.stop() in on_train_end() (#3103) 2023-06-08 19:16:51 +02:00
Glenn JocherandGitHub e88c89d269 Fix RayTune for < 8 CPUs (#3102) 2023-06-08 18:50:05 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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>
2023-06-07 08:17:39 +02:00
Glenn JocherandGitHub 8ac8ff72ae Fix code scanning alert - Incomplete URL substring sanitization (#3054) 2023-06-07 08:16:31 +02:00
LaughingandGitHub 31281400dd Simplify remove_zero_areas() function (#3060) 2023-06-07 08:13:56 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Burhan
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>
2023-06-06 01:09:27 +02:00
Zlobin VladimirGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn JocherAyush Chaurasia
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>
2023-06-06 00:29:12 +02:00
Ivan ShchekleinandGitHub 6057b267af Add DVC experiments logger with DVCLive (#2792) 2023-06-05 17:10:43 +02:00
Glenn JocherandGitHub 2b26572e42 Fix GlobalWheat2020.yaml auto download (#3007) 2023-06-05 01:37:36 +02:00
Glenn JocherandGitHub 2193ca1658 ultralytics 8.0.113 Adamax, NAdam, RAdam optimizers (#3004) 2023-06-04 22:38:25 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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