332 Commits

Author SHA1 Message Date
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
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 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
Sergiu Waxmann
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 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
Glenn Jocher
e70de6dacb
ultralytics 8.0.115 RayTune and DVC fixes (#3115) 2023-06-09 11:21:03 +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
Glenn Jocher
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 Jocher
8ac8ff72ae
Fix code scanning alert - Incomplete URL substring sanitization (#3054) 2023-06-07 08:16:31 +02:00
Laughing
31281400dd
Simplify remove_zero_areas() function (#3060) 2023-06-07 08:13:56 +02:00
Glenn Jocher
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 Vladimir
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 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
2193ca1658
ultralytics 8.0.113 Adamax, NAdam, RAdam optimizers (#3004) 2023-06-04 22:38:25 +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
Laughing
bb5bbc2893
Fix keypoints.conf update Results docs (#2977) 2023-06-04 22:33:17 +02:00
crbrz
a9129fb40e
Fix workspace and verbose arguments in TensorRT export (#2954)
Co-authored-by: crbrz <cristiab@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-06-02 16:06:01 +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
Glenn Jocher
d07ba25dc4
ultralytics 8.0.112 augment, results and docs updates (#2944) 2023-06-01 01:03:11 +02:00
Ayush Chaurasia
52ae65ac3b
Optionally train_args on attempt_load_weights() (#2928)
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-01 01:02:16 +02:00
Glenn Jocher
96fb5ee78e
Update license and tasks docs (#2941) 2023-06-01 00:42:55 +02:00
Laughing
c050b2d1a8
Task augment (#2924)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-06-01 00:41:10 +02:00
Laughing
f4b34fc30b
Fix save_txt in track mode and add Keypoints and Probs (#2921)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-06-01 00:40:10 +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
Glenn Jocher
90ec434ebb
Update stream_loaders.py to Path().absolute() (#2915) 2023-06-01 00:32:53 +02:00
Glenn Jocher
fd94d312da
ultralytics 8.0.111 refactored model.loss() method (#2911)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Snyk bot <snyk-bot@snyk.io>
2023-05-30 20:23:30 +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
Ayush Chaurasia
7f077f7654
Import ultralytics.nn.tasks as ultralytics.nn (#2889)
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-30 19:23:39 +02:00
tekert
50c9731808
Refactor best YouTube URL and add yt_dlp option (#2901)
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-30 19:13:03 +02:00