Commit Graph

  • 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
  • f56c9bcc26 Segmentation support & other enchancements (#40) Ayush Chaurasia 2022-11-08 20:57:57 +05:30
  • c617ee1c79 Fix dataloader2 (#35) Laughing 2022-11-08 05:47:25 -06:00
  • 523eff99e2 Update ci.yaml (#37) Glenn Jocher 2022-11-07 19:27:55 +01:00
  • a733e41bcc Bump actions/setup-python from 3 to 4 (#39) dependabot[bot] 2022-11-07 19:11:14 +01:00
  • 06dd49d862 Create dependabot.yml (#38) Glenn Jocher 2022-11-07 19:04:53 +01:00
  • e4f7458d90 Hydra *.yml extension deprecated fix (#34) Glenn Jocher 2022-11-07 00:52:08 +01:00
  • 6fe8bead35 Classify training cleanup (#33) Glenn Jocher 2022-11-07 00:15:57 +01:00
  • 2e9b18ce4e Fix dataloader (#32) Laughing 2022-11-04 03:00:40 -05:00
  • 92c60758dd Smart Model loading (#31) Ayush Chaurasia 2022-11-01 04:22:12 +05:30
  • 1054819a59 Add initial model interface (#30) Ayush Chaurasia 2022-10-26 01:21:15 +05:30
  • 7b560f7861 Model builder (#29) Ayush Chaurasia 2022-10-19 02:44:23 +05:30
  • c5cb76b356 Metrics and loss structure (#28) Ayush Chaurasia 2022-10-15 23:09:05 +05:30
  • d0b3c9812b Trainer + Dataloaders (#27) Ayush Chaurasia 2022-10-10 17:31:07 +05:30
  • 7a2e5fdfa3 Update ci.yaml (#21) Glenn Jocher 2022-09-21 13:57:28 +02:00
  • abbd177a78 Update README.md Glenn Jocher 2022-09-21 13:45:12 +02:00
  • f5dd65c709 Update README.md Glenn Jocher 2022-09-21 13:42:29 +02:00
  • 2472feafad Add missing setup.py fields (#18) Glenn Jocher 2022-09-19 15:49:09 +02:00
  • 171ad46d6b Create __init__.py (#17) Glenn Jocher 2022-09-19 15:19:13 +02:00
  • 419f6d7167 Add yolo package structure (#15) Ayush Chaurasia 2022-09-19 15:47:23 +05:30
  • 64212b67b9 tensorflow-macos comment (#14) Glenn Jocher 2022-09-18 17:14:46 +02:00
  • 3c58c26569 Update requirements.txt with fire>=0.4.0 (#12) Glenn Jocher 2022-09-17 11:41:08 +02:00
  • 088bd5b462 Create ci.yaml (#10) Glenn Jocher 2022-09-17 00:50:20 +02:00
  • 4aa74bb726 Normalize version with dev0 (#11) Glenn Jocher 2022-09-17 00:06:11 +02:00
  • 745f8e3759 Update requirements.txt (#9) Glenn Jocher 2022-09-16 23:42:08 +02:00
  • 39e9f3cae9 Initial pip structure (#1) Ayush Chaurasia 2022-09-17 02:13:06 +05:30
  • 95f6fa16d5 Initial commit Glenn Jocher 2022-09-11 19:39:46 +03:00