373 Commits

Author SHA1 Message Date
Sergiu Waxmann
2f02d8ea53
Docs updates for HUB, YOLOv4, YOLOv7, NAS (#3174)
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-15 21:17:10 +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
Glenn Jocher
095b856e75
Add inference API to CI (#3132)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-06-11 19:47:01 +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
Semih Demirel
a7c249e9c1
[Example] YOLOv8-ONNXRuntime (#2992)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
2023-06-05 18:57:03 +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
Glenn Jocher
0d47d11393
Fix DDP classify train imgsz (#2897) 2023-05-29 19:23:19 +02:00
Glenn Jocher
a2bb42dfe9
Fix Conv2() fusing (#2885) 2023-05-29 02:08:45 +02:00
Glenn Jocher
882dbe62ad
Update security policy (#2884) 2023-05-29 01:52:52 +02:00
Glenn Jocher
5baaf91b5b
Add ProfileModel min_time=60s argument (#2881) 2023-05-28 20:07:17 +02:00
Glenn Jocher
6bf5000a88
ultralytics 8.0.110 new profile and fraction train args (#2880) 2023-05-28 18:20:41 +02:00
Afif Al Mamun
75012e575e
Rename predictor postprocess orig_img to orig_imgs (#2877)
Signed-off-by: Afif Al Mamun <afifanik@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-05-28 17:54:05 +02:00
Glenn Jocher
dfe7f627d8
Remove Sentry from requirements (#2878) 2023-05-28 17:40:59 +02:00
John-Henry Lim
1a26e1e195
kwargs override cfg argument in model.train (#2839) (#2853)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-05-28 16:36:47 +02:00
setwgx
0c62f85750
Fix KeyError:line_thickness (#2852)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-05-28 16:23:10 +02:00
Laughing
ef1eff8830
Fix numpy FP64 bug in augmentation (#2841)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-05-28 16:21: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