Commit Graph

  • 58bccb1a9f Add CLI support for SAM, RTDETR (#3253) Glenn Jocher 2023-06-18 22:47:33 +02:00
  • 4c2033d7c3 Uninstall wandb from notebook environments (#3247) Glenn Jocher 2023-06-18 16:09:46 +02:00
  • 064d48eff9 ultralytics 8.0.119 RTDETR Trainer and ViT Docs (#3236) Glenn Jocher 2023-06-18 03:52:08 +02:00
  • bba7cf55a0 Fix data.yaml absolute path bug (#3242) Glenn Jocher 2023-06-18 03:35:24 +02:00
  • 9d38d2b044 Fix Classify confusion matrix (#3241) Glenn Jocher 2023-06-18 03:07:45 +02:00
  • 9c44560ec4 Return save_dir with prediction results (#3240) Glenn Jocher 2023-06-18 02:34:31 +02:00
  • 15c90bd404 Default classify training to pretrained=True (#3239) Glenn Jocher 2023-06-18 02:31:05 +02:00
  • e78fb683f4 default.yaml type comments (#3237) Glenn Jocher 2023-06-18 01:24:59 +02:00
  • 21df296425 Fix PIL getsize() deprecation warning (#3234) Glenn Jocher 2023-06-17 20:10:47 +02:00
  • c8441a5422 Fix scipy.ndimage.filters deprecation warning (#3233) Glenn Jocher 2023-06-17 19:53:41 +02:00
  • d3ca3e4b9b Fix Pose val plotting bug (#3232) Glenn Jocher 2023-06-17 19:35:19 +02:00
  • 1e5702a5b5 Add Ultralytics ViT Docs (#3230) Glenn Jocher 2023-06-17 16:44:41 +02:00
  • bd0f7ecf6f Docstrings arguments cleanup (#3229) Glenn Jocher 2023-06-17 16:17:07 +02:00
  • 62916b3b0a docs: add docs for hub datasets (#3212) Sergiu Waxmann 2023-06-17 14:05:41 +02:00
  • a0ba8ef5f0 Add RTDETR Trainer (#2745) Ayush Chaurasia 2023-06-17 17:16:18 +05:30
  • 03bce07848 Build reference for missing __init__.py files (#3227) Glenn Jocher 2023-06-17 13:31:46 +02:00
  • 0dfafaf39b Fix batch=-1 for exports (#3217) Glenn Jocher 2023-06-17 01:07:24 +02:00
  • d69a1e8046 Update save_dir on new predict args (#3215) Glenn Jocher 2023-06-16 19:59:46 +02:00
  • 431fad6834 Update Dockerfile-jetson (#3209) Glenn Jocher 2023-06-16 18:06:58 +02:00
  • d6e2ec1483 Jetson Docker fix opencv-python>=4.5.0 (#3208) Glenn Jocher 2023-06-16 15:10:36 +02:00
  • a5a34af2f4 ultralytics 8.0.118 updated Docs, OS and DDP fixes (#3197) Glenn Jocher 2023-06-16 02:30:58 +02:00
  • a0fa99c4db Comet Updates (#3186) Dhruv Nair 2023-06-16 05:36:28 +05:30
  • 0d91d6df6e Add world_size check before setting up DDP train (#3191) Bruno Arine 2023-06-15 21:05:15 -03:00
  • f8e1dcc43f Fix operating system to list (#3196) Glenn Jocher 2023-06-16 02:04:31 +02:00
  • 0a229bd40f Add patches section and fix Docs CI (#3193) Glenn Jocher 2023-06-15 21:44:25 +02:00
  • 2f02d8ea53 Docs updates for HUB, YOLOv4, YOLOv7, NAS (#3174) Sergiu Waxmann 2023-06-15 21:17:10 +02:00
  • c340f84ce9 ultralytics 8.0.117 NAS export, classify and tasks banner URL fixes (#3145) Glenn Jocher 2023-06-12 20:14:46 +02:00
  • b59342b81c Fix platform set to tuple (#3136) Laughing 2023-06-12 16:35:37 +08:00
  • f4e8b39fc2 ultralytics 8.0.116 NAS, DVC, YOLOv5u updates (#3124) Glenn Jocher 2023-06-11 20:39:32 +02:00
  • 095b856e75 Add inference API to CI (#3132) Glenn Jocher 2023-06-11 19:47:01 +02:00
  • 9e859c949e Use model_info_for_loggers() in DVC logger (#3127) Ivan Shcheklein 2023-06-11 08:59:42 -07:00
  • 8e60fc9276 Add device=cuda support (#3133) Glenn Jocher 2023-06-11 17:58:10 +02:00
  • ff91fbd9c3 Add Ultralytics tasks and YOLO-NAS models (#2735) Glenn Jocher 2023-06-10 18:54:14 +02:00
  • e70de6dacb ultralytics 8.0.115 RayTune and DVC fixes (#3115) Glenn Jocher 2023-06-09 11:21:03 +02:00
  • 3892e38021 fix(loggers): DVCLive add package version check (#3112) Ivan Shcheklein 2023-06-09 02:19:34 -07:00
  • 29c954a138 Remove Neptune run.stop() in on_train_end() (#3103) Siddhant Sadangi 2023-06-08 19:16:51 +02:00
  • e88c89d269 Fix RayTune for < 8 CPUs (#3102) Glenn Jocher 2023-06-08 18:50:05 +02:00
  • 3ae81ee9d1 Update val max_dets=args.max_det=300 (#3051) Glenn Jocher 2023-06-07 08:17:39 +02:00
  • 8ac8ff72ae Fix code scanning alert - Incomplete URL substring sanitization (#3054) Glenn Jocher 2023-06-07 08:16:31 +02:00
  • 31281400dd Simplify remove_zero_areas() function (#3060) Laughing 2023-06-07 14:13:56 +08:00
  • 8940a27bdb ultralytics 8.0.114 automatic optimizer selection (#3037) Glenn Jocher 2023-06-06 01:09:27 +02:00
  • 01273c5ca1 Add rt info for OpenVINO ModelAPI (#2891) Zlobin Vladimir 2023-06-06 02:29:12 +04:00
  • a7c249e9c1 [Example] YOLOv8-ONNXRuntime (#2992) Semih Demirel 2023-06-05 19:57:03 +03:00
  • 6057b267af Add DVC experiments logger with DVCLive (#2792) Ivan Shcheklein 2023-06-05 05:10:43 -10:00
  • 2b26572e42 Fix GlobalWheat2020.yaml auto download (#3007) Glenn Jocher 2023-06-05 01:37:36 +02:00
  • 2193ca1658 ultralytics 8.0.113 Adamax, NAdam, RAdam optimizers (#3004) Glenn Jocher 2023-06-04 22:38:25 +02:00
  • 451cf8b647 Add Adamax, NAdam, RAdam optimizers (#2969) Glenn Jocher 2023-06-04 22:35:50 +02:00
  • f502b50365 fix overlap_mask in SegmentationLoss (#2985) Laughing 2023-06-05 04:35:33 +08:00
  • bb5bbc2893 Fix keypoints.conf update Results docs (#2977) Laughing 2023-06-05 04:33:17 +08:00
  • a9129fb40e Fix workspace and verbose arguments in TensorRT export (#2954) crbrz 2023-06-02 17:06:01 +03:00
  • 67cf53b475 Move check_amp() to checks.py (#2948) Ayush Chaurasia 2023-06-02 19:13:07 +05:30
  • d07ba25dc4 ultralytics 8.0.112 augment, results and docs updates (#2944) Glenn Jocher 2023-06-01 01:03:11 +02:00
  • 52ae65ac3b Optionally train_args on attempt_load_weights() (#2928) Ayush Chaurasia 2023-06-01 04:32:16 +05:30
  • 96fb5ee78e Update license and tasks docs (#2941) Glenn Jocher 2023-06-01 00:42:55 +02:00
  • c050b2d1a8 Task augment (#2924) Laughing 2023-06-01 06:41:10 +08:00
  • f4b34fc30b Fix save_txt in track mode and add Keypoints and Probs (#2921) Laughing 2023-06-01 06:40:10 +08:00
  • 6c65934b55 W&B updates (#2895) Ayush Chaurasia 2023-06-01 04:03:06 +05:30
  • 90ec434ebb Update stream_loaders.py to Path().absolute() (#2915) Glenn Jocher 2023-06-01 00:32:53 +02:00
  • fd94d312da ultralytics 8.0.111 refactored model.loss() method (#2911) Glenn Jocher 2023-05-30 20:23:30 +02:00
  • 305cde69d0 Update generate_ddp_file for improved overrides (#2909) Laughing 2023-05-31 01:41:44 +08:00
  • facb7861cf Move loss calculation to head (#2874) Ayush Chaurasia 2023-05-30 22:54:30 +05:30
  • 7f077f7654 Import ultralytics.nn.tasks as ultralytics.nn (#2889) Ayush Chaurasia 2023-05-30 22:53:39 +05:30
  • 50c9731808 Refactor best YouTube URL and add yt_dlp option (#2901) tekert 2023-05-30 14:13:03 -03:00
  • 0d47d11393 Fix DDP classify train imgsz (#2897) Glenn Jocher 2023-05-29 19:23:19 +02:00
  • a2bb42dfe9 Fix Conv2() fusing (#2885) Glenn Jocher 2023-05-29 02:08:45 +02:00
  • 882dbe62ad Update security policy (#2884) Glenn Jocher 2023-05-29 01:52:52 +02:00
  • 5baaf91b5b Add ProfileModel min_time=60s argument (#2881) Glenn Jocher 2023-05-28 20:07:17 +02:00
  • 6bf5000a88 ultralytics 8.0.110 new profile and fraction train args (#2880) Glenn Jocher 2023-05-28 18:20:41 +02:00
  • 75012e575e Rename predictor postprocess orig_img to orig_imgs (#2877) Afif Al Mamun 2023-05-28 21:54:05 +06:00
  • dfe7f627d8 Remove Sentry from requirements (#2878) Glenn Jocher 2023-05-28 17:40:59 +02:00
  • 1a26e1e195 kwargs override cfg argument in model.train (#2839) (#2853) John-Henry Lim 2023-05-28 22:36:47 +08:00
  • 0c62f85750 Fix KeyError:line_thickness (#2852) setwgx 2023-05-28 22:23:10 +08:00
  • ef1eff8830 Fix numpy FP64 bug in augmentation (#2841) Laughing 2023-05-28 22:21:59 +08:00
  • 527a97759b Revert loss head PR (#2873) Ayush Chaurasia 2023-05-28 19:45:41 +05:30
  • 6391c60089 New train profile argument for loggers (#2862) Glenn Jocher 2023-05-28 03:51:49 +02:00
  • 0bdd4ad379 New dataset fraction=1.0 argument (#2860) Glenn Jocher 2023-05-28 02:13:46 +02:00
  • 61fa5efe6d build_optimizer() assign all parameters (#2855) Glenn Jocher 2023-05-26 21:03:03 +02:00
  • 441e67d330 Add Conv2() module (#2820) Glenn Jocher 2023-05-25 18:12:50 +02:00
  • d19c5b6ce8 Move loss to task heads (#2825) Ayush Chaurasia 2023-05-25 16:07:54 +05:30
  • f23a03596d ultralytics 8.0.109 HUB training fix (#2818) Glenn Jocher 2023-05-25 06:44:31 +02:00
  • ffc0e8ccf7 ultralytics 8.0.108 add Meituan YOLOv6 models (#2811) Glenn Jocher 2023-05-25 00:43:32 +02:00
  • 07b57c03c8 Buffered Mosaic for reduced HDD reads (#2791) Glenn Jocher 2023-05-25 00:42:13 +02:00
  • dada5b73c4 ultralytics 8.0.107 (#2778) Glenn Jocher 2023-05-23 13:50:24 +02:00
  • 4db686a315 ultralytics 8.0.106 (#2736) Glenn Jocher 2023-05-22 13:31:19 +02:00
  • 23fc50641c ultralytics 8.0.105 classification hyp fix and new onplot callbacks (#2684) Glenn Jocher 2023-05-17 19:10:20 +02:00
  • b1119d512e ultralytics 8.0.104 bug fixes and thop dependency removal (#2665) Glenn Jocher 2023-05-17 02:26:01 +02:00
  • 7884098857 ultralytics 8.0.103 minor fixes (#2634) Glenn Jocher 2023-05-16 01:39:52 +02:00
  • e2e3e367a2 ultralytics 8.0.101 mosaic9() and loss bug fixes (#2608) Glenn Jocher 2023-05-15 12:12:34 +02:00
  • ff211f4037 fix loss (#2614) Laughing 2023-05-15 18:09:34 +08:00
  • dce4efce48 ultralytics 8.0.100 add Mosaic9() augmentation (#2605) Glenn Jocher 2023-05-14 20:43:35 +02:00
  • db1c5885d5 ultralytics 8.0.99 HUB resume fix and Docs updates (#2567) Glenn Jocher 2023-05-12 18:33:32 +02:00
  • 229119c376 ultralytics 8.0.98 add Baidu RT-DETR models (#2527) Glenn Jocher 2023-05-11 01:15:20 +02:00
  • d1107ca4cb ultralytics 8.0.97 confusion matrix, windows, docs updates (#2511) Glenn Jocher 2023-05-09 21:20:34 +02:00
  • 6ee3a9a74b ultralytics 8.0.96 TAL speed and memory improvements (#2484) Glenn Jocher 2023-05-08 23:41:27 +02:00
  • e21428ca4e ultralytics 8.0.94 HUBDatasetStats() Segment and Pose support (#2450) Glenn Jocher 2023-05-08 01:59:12 +02:00
  • af49a85cf3 Fix Jetson Docker (#2433) Glenn Jocher 2023-05-06 01:58:27 +02:00
  • ddb354ce5e ultralytics 8.0.93 HUB docs and JSON2YOLO converter (#2431) Glenn Jocher 2023-05-06 01:12:43 +02:00
  • 0ebd3f2959 ultralytics 8.0.92 updates and fixes (#2361) Glenn Jocher 2023-05-04 01:27:00 +02:00
  • 3fd317edfd ultralytics 8.0.91 tracker fix and docs comments (#2343) Glenn Jocher 2023-05-01 22:20:05 +02:00
  • 44c7c3514d ultralytics 8.0.90 actions and docs improvements (#2326) Glenn Jocher 2023-04-29 20:16:56 +02:00