Commit Graph

571 Commits

Author SHA1 Message Date
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
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
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
29c954a138 Remove Neptune run.stop() in on_train_end() (#3103) 2023-06-08 19:16:51 +02:00
e88c89d269 Fix RayTune for < 8 CPUs (#3102) 2023-06-08 18:50:05 +02:00
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
8ac8ff72ae Fix code scanning alert - Incomplete URL substring sanitization (#3054) 2023-06-07 08:16:31 +02:00
31281400dd Simplify remove_zero_areas() function (#3060) 2023-06-07 08:13:56 +02:00
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
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
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
6057b267af Add DVC experiments logger with DVCLive (#2792) 2023-06-05 17:10:43 +02:00
2b26572e42 Fix GlobalWheat2020.yaml auto download (#3007) 2023-06-05 01:37:36 +02:00
2193ca1658 ultralytics 8.0.113 Adamax, NAdam, RAdam optimizers (#3004) 2023-06-04 22:38:25 +02:00
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
f502b50365 fix overlap_mask in SegmentationLoss (#2985) 2023-06-04 22:35:33 +02:00
bb5bbc2893 Fix keypoints.conf update Results docs (#2977) 2023-06-04 22:33:17 +02:00
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
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
d07ba25dc4 ultralytics 8.0.112 augment, results and docs updates (#2944) 2023-06-01 01:03:11 +02:00
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
96fb5ee78e Update license and tasks docs (#2941) 2023-06-01 00:42:55 +02:00
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
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
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
90ec434ebb Update stream_loaders.py to Path().absolute() (#2915) 2023-06-01 00:32:53 +02:00
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
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
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
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
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
0d47d11393 Fix DDP classify train imgsz (#2897) 2023-05-29 19:23:19 +02:00
a2bb42dfe9 Fix Conv2() fusing (#2885) 2023-05-29 02:08:45 +02:00
882dbe62ad Update security policy (#2884) 2023-05-29 01:52:52 +02:00
5baaf91b5b Add ProfileModel min_time=60s argument (#2881) 2023-05-28 20:07:17 +02:00
6bf5000a88 ultralytics 8.0.110 new profile and fraction train args (#2880) 2023-05-28 18:20:41 +02:00
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
dfe7f627d8 Remove Sentry from requirements (#2878) 2023-05-28 17:40:59 +02:00
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
0c62f85750 Fix KeyError:line_thickness (#2852)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-05-28 16:23:10 +02:00
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
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
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
0bdd4ad379 New dataset fraction=1.0 argument (#2860) 2023-05-28 02:13:46 +02:00
61fa5efe6d build_optimizer() assign all parameters (#2855)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-05-26 21:03:03 +02:00