Commit Graph

  • 0e5a7ae623
    General trainer cleanup (#147) Ayush Chaurasia 2023-01-07 19:25:48 +05:30
  • f8a13c49a0
    Model saved_model export (#151) Glenn Jocher 2023-01-06 21:44:44 +01:00
  • d17d1e064d
    Add Comet integration (#153) Ayush Chaurasia 2023-01-06 01:38:57 +05:30
  • 314da263c7
    Default confidences predict 0.25, val 0.001 (#149) Glenn Jocher 2023-01-05 15:56:52 +01:00
  • 3cbf3ec455
    Global settings typechecking (#148) Glenn Jocher 2023-01-05 15:13:22 +01:00
  • 19334ebb16
    Clean validator (#144) Laughing 2023-01-05 21:14:01 +08:00
  • 46cb657b64
    Merge model() and model.predict() (#146) Glenn Jocher 2023-01-05 14:13:29 +01:00
  • 99275814f1
    Add AutoBatch from YOLOv5 (#145) Glenn Jocher 2023-01-05 13:49:16 +01:00
  • 172cef2d20
    CoreML NMS and half fixes (#143) Glenn Jocher 2023-01-05 00:20:54 +01:00
  • 55bdca6768
    Improvements (#142) Glenn Jocher 2023-01-04 16:37:46 +01:00
  • dcd8ef68e6
    Add colab template and minor fixes (#137) Ayush Chaurasia 2023-01-04 02:00:39 +05:30
  • 3c18aa5278
    Simplify argument names (#141) Glenn Jocher 2023-01-03 19:47:56 +01:00
  • d74de2582c
    Pip debug fixes (#139) Glenn Jocher 2023-01-03 18:54:44 +01:00
  • 340376f7a6
    Fix resume (#138) Laughing 2023-01-03 21:06:22 +08:00
  • 82c849c163
    Fix load and resume and update autodownload endpoint (#136) Ayush Chaurasia 2023-01-03 01:42:01 +05:30
  • 6d5123297e
    Fix CLI detect and segment resume (#134) Ayush Chaurasia 2023-01-03 00:31:17 +05:30
  • c5c86a3acd
    CLI DDP fixes (#135) Glenn Jocher 2023-01-02 19:55:04 +01:00
  • 8f3cd52844
    Fix yolo mode=train CLI bug on model load (#133) Glenn Jocher 2023-01-02 19:24:44 +01:00
  • c3d961fb03
    Unified model loading with backwards compatibility (#132) Glenn Jocher 2023-01-02 17:37:23 +01:00
  • 8996c5c6cf
    [Docs]: Link buttons, add autobackend, BaseModel and ops (#130) Ayush Chaurasia 2023-01-02 20:42:30 +05:30
  • af6e3c536b
    Override fixes and general updates (#129) Glenn Jocher 2023-01-02 15:55:32 +01:00
  • d76d7af566 Simplify usage, remove YOLO.new() and YOLO.load() Glenn Jocher 2023-01-02 15:42:01 +01:00
  • 7b21a87da3
    HUB initial setup (#108) Ayush Chaurasia 2023-01-02 00:51:37 +05:30
  • 2bc9a5c87e
    HUB setup (#108) Ayush Chaurasia 2023-01-02 00:51:14 +05:30
  • c6eb6720de
    Add pred, export and val callbacks (#126) Ayush Chaurasia 2023-01-01 22:46:10 +05:30
  • 63c7a74691
    Replace nosave and noval with save and val (#127) Glenn Jocher 2023-01-01 17:50:06 +01:00
  • 598f17a472
    Add global settings.yaml in USER_CONFIG_DIR (#125) Glenn Jocher 2022-12-31 21:40:41 +01:00
  • a9b9fe7618
    Add Dockerfiles and update Docs README (#124) Glenn Jocher 2022-12-31 18:39:42 +01:00
  • df4fc14c10
    Docstring additions (#122) Glenn Jocher 2022-12-31 13:42:45 +01:00
  • c9f3e469cb
    Add CoreML iOS updates (#121) Glenn Jocher 2022-12-30 21:33:43 +01:00
  • fec13ec773
    Add YOLO8x6 and YAML syntax improvements (#120) Glenn Jocher 2022-12-30 13:32:08 +01:00
  • 076d73cfaa
    Create Exporter() Class (#117) Glenn Jocher 2022-12-30 01:28:41 +01:00
  • a9dc1637c2
    Mask pycocotools (#116) Laughing 2022-12-30 01:01:03 +08:00
  • 37e0f6e0e4
    Import DEFAULT_CONFIG from utils/__init__.py (#115) Glenn Jocher 2022-12-29 14:58:03 +01:00
  • 92dad1c1b5
    Start export implementation (#110) Glenn Jocher 2022-12-29 14:17:14 +01:00
  • c1b38428bc
    Update save_dir rank check (#114) Ayush Chaurasia 2022-12-29 17:32:01 +05:30
  • 401bc15345
    fix cache in new loader (#112) Laughing 2022-12-29 17:36:28 +08:00
  • 100e21ee80
    fix multiple dirs in DDP mode (#113) Ayush Chaurasia 2022-12-29 15:02:17 +05:30
  • 34829a6b29
    Make YOLO a module (#111) Ayush Chaurasia 2022-12-29 00:08:37 +05:30
  • 0303ced8ab
    Model enhancements 2 (#109) Ayush Chaurasia 2022-12-28 21:01:51 +05:30
  • 8406b49b49
    Integration of v8 segmentation (#107) Laughing 2022-12-28 23:01:38 +08:00
  • 384f0ef1c6
    Model interface enhancement (#106) Ayush Chaurasia 2022-12-28 18:05:01 +05:30
  • 38d6df55cb
    Fix save_json(predn, batch) (#105) Glenn Jocher 2022-12-27 18:42:36 +01:00
  • 8b6466f731
    Move tests outside package (#104) Ayush Chaurasia 2022-12-27 22:16:07 +05:30
  • 8028e2b1b8
    Make optimizer static method inside trainer (#103) Ayush Chaurasia 2022-12-27 21:32:03 +05:30
  • 5c6d11bdb2
    Allocated updated pycocotools metrics fix (#101) Laughing 2022-12-27 22:49:13 +08:00
  • 441126a77e
    Add l/x models (#100) Glenn Jocher 2022-12-27 05:06:15 +01:00
  • 6f0ba81427
    Add best.pt val and COCO pycocotools val (#98) Glenn Jocher 2022-12-27 04:56:24 +01:00
  • a1808eeda4
    logger updates (#97) Ayush Chaurasia 2022-12-26 22:42:17 +05:30
  • 48cffa176e
    Shorten module paths with new 'nn' dir (#96) Glenn Jocher 2022-12-26 17:34:28 +01:00
  • 4fb04be20b
    DDP and new dataloader Fix (#95) Ayush Chaurasia 2022-12-26 20:05:49 +05:30
  • 16e3c08883
    Import YOLOv5 dataloader (#94) Ayush Chaurasia 2022-12-26 06:00:01 +05:30
  • ae05d44877
    Simplify augmentations (#93) Glenn Jocher 2022-12-25 21:21:26 +01:00
  • 249dfbdc05
    Fix ClearML Mosaic callback to 'on_train_epoch_end' (#92) Glenn Jocher 2022-12-25 17:04:24 +01:00
  • 0298821467
    ClearML: Mosaic Logging (#91) Ayush Chaurasia 2022-12-25 19:06:41 +05:30
  • 98815d560f
    YOLOv5 updates (#90) Glenn Jocher 2022-12-25 14:33:18 +01:00
  • ebd3cfb2fd
    YOLOv8 architecture updates from R&D branch (#88) Glenn Jocher 2022-12-24 18:10:44 +01:00
  • 5fbea25f0b
    Use integrations only in master process (#89) Ayush Chaurasia 2022-12-24 22:09:30 +05:30
  • cb4f20f3cf
    Add TensorBoard support (#87) Glenn Jocher 2022-12-24 14:37:46 +01:00
  • 248d54ca03
    Update metrics names (#85) Glenn Jocher 2022-12-24 02:32:24 +01:00
  • 6432afc5f9
    Rename img_size to imgsz (#86) Glenn Jocher 2022-12-24 00:39:09 +01:00
  • ae2443c210
    Add flops, num_params, inference speed logging and best.pt logging (#84) Ayush Chaurasia 2022-12-24 00:44:21 +05:30
  • f0fff8c13e
    Simplify cli.py and fix Detect train Usage (#83) Glenn Jocher 2022-12-22 15:22:16 +01:00
  • 74feef30c4
    Update cache_version = 1.0 (#82) Glenn Jocher 2022-12-22 12:44:28 +01:00
  • 40f7e0806e
    Add v8 modules (#81) Ayush Chaurasia 2022-12-21 04:59:58 +05:30
  • d198f3f9d1
    Update docs (#73) Ayush Chaurasia 2022-12-20 09:53:55 +05:30
  • 681cfc1c35
    Make config overrides user friendly (#80) Ayush Chaurasia 2022-12-20 09:50:08 +05:30
  • 7690cae2fc
    add a naive DDP for model interface (#78) Laughing 2022-12-19 13:24:39 +08:00
  • 48c95ba083
    Change class depending on dataset in model interface (#77) Ayush Chaurasia 2022-12-16 13:56:25 +05:30
  • 24a7c068ad
    update docs (#76) Ayush Chaurasia 2022-12-15 11:48:03 +05:30
  • eb5adf4e0b
    Model enhancement (#75) Ayush Chaurasia 2022-12-14 14:33:31 +05:30
  • d85b44f259
    Update docs (#71) Ayush Chaurasia 2022-12-12 09:21:00 +05:30
  • e629335f6d
    Revert augment_hyps (#70) Laughing 2022-12-09 04:19:53 -06:00
  • d63ee112d4
    General cleanup (#69) Laughing 2022-12-08 08:28:13 -06:00
  • 7ae45c6cc4
    [WIP] Model interface (#68) Ayush Chaurasia 2022-12-08 13:22:56 +05:30
  • e6737f1207
    Predictor support (#65) Ayush Chaurasia 2022-12-07 10:33:10 +05:30
  • 479992093c
    Flake8 updates (#66) Glenn Jocher 2022-12-06 15:09:53 -08:00
  • fbeeb5d1e1
    add resuming (#63) Laughing 2022-12-05 20:56:41 -06:00
  • de3e6ca54d
    Fix CITATION.cff typos (#64) Glenn Jocher 2022-12-05 12:17:25 -08:00
  • 6903ccdf6b
    Create CITATION.cff (#62) Glenn Jocher 2022-12-05 12:14:59 -08:00
  • fe75a9ce67
    docs setup (#61) Ayush Chaurasia 2022-12-05 06:04:57 +05:30
  • 7ec7cf3aef
    Detection support (#60) Ayush Chaurasia 2022-12-03 21:01:23 +05:30
  • 5a52e7663a
    standalone val (#56) Ayush Chaurasia 2022-11-30 15:04:44 +05:30
  • 3a241e4cea
    update segment training (#57) Laughing 2022-11-29 05:30:08 -06:00
  • d0b0fe2592
    CLI updates (#58) Ayush Chaurasia 2022-11-29 15:41:37 +05:30
  • c5f5b80c04
    Deterministic training (#53) Ayush Chaurasia 2022-11-25 02:06:14 +05:30
  • 793dde365d Revert "update" ayush chaurasia 2022-11-25 01:54:50 +05:30
  • 486bc49414 update ayush chaurasia 2022-11-25 01:11:55 +05:30
  • d7df1770fa
    Add warmup and accumulation (#52) Ayush Chaurasia 2022-11-25 00:42:16 +05:30
  • 298287298d
    Add clearml logging (#51) Ayush Chaurasia 2022-11-24 19:55:03 +05:30
  • 512a225ce8
    Cli support (#50) Ayush Chaurasia 2022-11-23 16:07:27 +05:30
  • 4291b9c31c
    Add EMA and model checkpointing (#49) Ayush Chaurasia 2022-11-19 23:37:26 +05:30
  • 27d6545117
    General console printout updates (#48) Glenn Jocher 2022-11-19 16:08:16 +01:00
  • 8530e3fae0
    Logging fix from YOLOv5 (#47) Glenn Jocher 2022-11-19 14:51:52 +01:00
  • 47f1cb3ef4
    Fix some cuda training issues of segmentation (#46) Laughing 2022-11-17 06:44:02 -06:00
  • db1031a1a9
    Allow setting model attributes before training (#45) Ayush Chaurasia 2022-11-17 10:35:53 +05:30
  • 832ea56eb4
    update model initialization design, supports custom data/num_classes (#44) Ayush Chaurasia 2022-11-15 20:06:29 +05:30
  • 1f3aad86c1
    new check_dataset functions (#43) Ayush Chaurasia 2022-11-12 19:32:41 +05:30
  • d143ac666f
    [rename] - preprocess-batch -> preprocess, preprocess_preds -> postprocess (#42) Ayush Chaurasia 2022-11-09 22:05:15 +05:30
  • 4c68b9dcf6
    Add is_colab() and is_kaggle() (#41) Glenn Jocher 2022-11-08 17:14:49 +01:00