Commit Graph

  • 5e38c7d71b
    Improve NMS speed (#3467) Andy 2023-07-03 16:20:49 +02:00
  • 586c95b8a9
    Add torch.Tensor results image saving (#3475) Neloy262 2023-07-03 20:08:02 +06:00
  • cd0bf05aa4
    Cap head scaling with min(nc, 100) (#3436) Glenn Jocher 2023-06-28 14:11:35 +02:00
  • 17e139220b
    ultralytics 8.0.124 lapx package and multiple-unzip fix (#3423) Glenn Jocher 2023-06-28 00:36:01 +02:00
  • 27a24b2796
    Change key format for images dict in convert_coco (#3419) (#3420) Liam Parker 2023-06-27 14:01:26 -05:00
  • 05760b04c4
    Fix dataset multiple unzip behavior (#3413) Glenn Jocher 2023-06-27 17:14:39 +02:00
  • 42057fdc95
    lapx package replacement for lap (#3383) Glenn Jocher 2023-06-26 21:30:14 +02:00
  • 12f3089110
    Update Discord invite link (#3397) Glenn Jocher 2023-06-26 19:13:02 +02:00
  • e93a5fbff6
    Fix RTDETR val_batch_pred (#3392) Laughing 2023-06-26 19:37:33 +08:00
  • 4e08e12256
    Docker apt install gnupg g++ for lap (#3381) Glenn Jocher 2023-06-26 00:54:37 +02:00
  • b5a2be8ca8
    ultralytics 8.0.123 Ubuntu security and VideoWriter codec fixes (#3380) Glenn Jocher 2023-06-25 23:59:28 +02:00
  • 8b50b2a897
    Fix video writing (#3379) Glenn Jocher 2023-06-25 23:11:00 +02:00
  • 9b00d37022
    FROM ubuntu:lunar-20230615 and lap install fixes (#3374) Glenn Jocher 2023-06-25 21:05:38 +02:00
  • ca2c8774d8
    Add FastSAM docs page (#3371) Glenn Jocher 2023-06-25 14:08:19 +02:00
  • 15c027a9fc
    Add torch.Tensor checks and pip badges (#3368) Glenn Jocher 2023-06-25 02:52:00 +02:00
  • 682c9ef70f
    ultralytics 8.0.122 Fix torch.Tensor inference (#3363) Glenn Jocher 2023-06-25 01:36:07 +02:00
  • 51d8cfa9c3
    Fix RT-DETR exported onnx model (#3317) Kayzwer 2023-06-24 22:34:24 +08:00
  • 2f58b5821a
    [Example] Add example application for .NET (#3311) Compunet 2023-06-24 16:57:59 +03:00
  • 5381fc8a58
    Warn save disabled for torch inference (#3361) Glenn Jocher 2023-06-24 15:57:17 +02:00
  • 2ebd808b69
    Increase NCCL timeout from 1 hour to 3 hours (#3343) Jieun Park 2023-06-24 22:56:18 +09:00
  • 137552996a
    ultralytics 8.0.121 RTDETR fix and Docs updates (#3314) Glenn Jocher 2023-06-22 00:58:37 +02:00
  • 35c02c8319
    Fix Pose export from YAML (#3312) Glenn Jocher 2023-06-21 18:04:22 +02:00
  • 9d1e5567de
    [RTDETR]Fix val loss (#3280) Laughing 2023-06-20 23:59:04 +08:00
  • d8701b42ca
    ultralytics 8.0.120 CLI support for SAM, RTDETR (#3273) Glenn Jocher 2023-06-19 22:12:48 +02:00
  • 2e8313b032
    Allow setting MLFlow experiment name through env var (#3266) Jeroen 2023-06-19 22:09:17 +02:00
  • 0632c88722
    Update datasets/classify/index.md Docs (#3244) Wang Xin 2023-06-20 04:06:47 +08:00
  • e9205974ff
    Docs: add docs for HUB models (#3270) Sergiu Waxmann 2023-06-19 21:55:30 +02:00
  • eda7758521
    [RTDETR]fix "Grad strides do not match bucket view strides" when training with DDP (#3255) Laughing 2023-06-19 22:09:06 +08:00
  • 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