Commit Graph

345 Commits

Author SHA1 Message Date
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
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
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
96fb5ee78e Update license and tasks docs (#2941) 2023-06-01 00:42:55 +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
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
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
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
d19c5b6ce8 Move loss to task heads (#2825)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
2023-05-25 16:07:54 +05:30
f23a03596d ultralytics 8.0.109 HUB training fix (#2818)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-05-25 06:44:31 +02:00
ffc0e8ccf7 ultralytics 8.0.108 add Meituan YOLOv6 models (#2811)
Co-authored-by: Michael Currie <mcurrie@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Hicham Talaoubrid <98521878+HichTala@users.noreply.github.com>
Co-authored-by: Zlobin Vladimir <vladimir.zlobin@intel.com>
Co-authored-by: Szymon Mikler <sjmikler@gmail.com>
2023-05-25 00:43:32 +02:00
07b57c03c8 Buffered Mosaic for reduced HDD reads (#2791)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-05-25 00:42:13 +02:00
dada5b73c4 ultralytics 8.0.107 (#2778)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Peter van Lunteren <contact@pvanlunteren.com>
2023-05-23 13:50:24 +02:00
4db686a315 ultralytics 8.0.106 (#2736)
Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: vyskocj <whiskey1939@seznam.cz>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: triple Mu <gpu@163.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
2023-05-22 13:31:19 +02:00
23fc50641c ultralytics 8.0.105 classification hyp fix and new onplot callbacks (#2684)
Co-authored-by: ayush chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Ivan Shcheklein <shcheklein@gmail.com>
2023-05-17 19:10:20 +02:00
b1119d512e ultralytics 8.0.104 bug fixes and thop dependency removal (#2665)
Co-authored-by: Kevin Abraham <5976139+abraha2d@users.noreply.github.com>
Co-authored-by: Kevin Abraham <abraha2d@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
2023-05-17 02:26:01 +02:00
7884098857 ultralytics 8.0.103 minor fixes (#2634)
Co-authored-by: kssion <appplus@icloud.com>
Co-authored-by: Eli Spizzichino <diramazioni@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: xbkaishui <xxkaishui@gmail.com>
2023-05-16 01:39:52 +02:00
e2e3e367a2 ultralytics 8.0.101 mosaic9() and loss bug fixes (#2608)
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
2023-05-15 12:12:34 +02:00
ff211f4037 fix loss (#2614) 2023-05-15 12:09:34 +02:00
dce4efce48 ultralytics 8.0.100 add Mosaic9() augmentation (#2605)
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: Tommy in Tongji <36354458+TommyZihao@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: BIGBOSS-FOX <47949596+BIGBOSS-FOX@users.noreply.github.com>
Co-authored-by: xbkaishui <xxkaishui@gmail.com>
2023-05-14 20:43:35 +02:00
db1c5885d5 ultralytics 8.0.99 HUB resume fix and Docs updates (#2567)
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
2023-05-12 18:33:32 +02:00
229119c376 ultralytics 8.0.98 add Baidu RT-DETR models (#2527)
Co-authored-by: Kalen Michael <kalenmike@gmail.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: Dowon <ks2515@naver.com>
2023-05-11 01:15:20 +02:00