Commit Graph
56 Commits
Author SHA1 Message Date
d387359f74 [Docs]: Add customization tutorial and address feedback (#155)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-01-08 14:01:22 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
07eab49c3d Add Classification model YAML support (#154)
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-01-07 20:04:34 +01:00
Ayush ChaurasiaGitHubGlenn Jocherpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Laughing
0e5a7ae623 General trainer cleanup (#147)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.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-01-07 19:25:48 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
d17d1e064d Add Comet integration (#153)
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-01-05 21:08:57 +01:00
Ayush ChaurasiaGitHubGlenn Jocherpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
dcd8ef68e6 Add colab template and minor fixes (#137)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-03 21:30:39 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
82c849c163 Fix load and resume and update autodownload endpoint (#136)
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-01-02 21:12:01 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
6d5123297e Fix CLI detect and segment resume (#134)
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-01-02 20:01:17 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
8996c5c6cf [Docs]: Link buttons, add autobackend, BaseModel and ops (#130)
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-01-02 20:42:30 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
7b21a87da3 HUB initial setup (#108)
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-01-01 20:21:37 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
2bc9a5c87e HUB setup (#108)
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-01-01 20:21:14 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
c6eb6720de Add pred, export and val callbacks (#126)
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-01-01 18:16:10 +01:00
c1b38428bc Update save_dir rank check (#114)
Co-authored-by: Laughing-q <1185102784@qq.com>
2022-12-29 13:02:01 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Laughing-q
100e21ee80 fix multiple dirs in DDP mode (#113)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
2022-12-29 15:02:17 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
34829a6b29 Make YOLO a module (#111)
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>
2022-12-28 19:38:37 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
0303ced8ab Model enhancements 2 (#109)
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>
2022-12-28 21:01:51 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
384f0ef1c6 Model interface enhancement (#106)
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>
2022-12-28 13:35:01 +01:00
Ayush ChaurasiaandGitHub 8b6466f731 Move tests outside package (#104) 2022-12-27 22:16:07 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
8028e2b1b8 Make optimizer static method inside trainer (#103)
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>
2022-12-27 17:02:03 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
a1808eeda4 logger updates (#97)
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>
2022-12-26 18:12:17 +01:00
Ayush ChaurasiaGitHubLaughing-qGlenn Jocherpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
4fb04be20b DDP and new dataloader Fix (#95)
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-26 15:35:49 +01:00
Ayush ChaurasiaGitHubGlenn Jocherpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
16e3c08883 Import YOLOv5 dataloader (#94)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-26 01:30:01 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
0298821467 ClearML: Mosaic Logging (#91)
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>
2022-12-25 14:36:41 +01:00
5fbea25f0b Use integrations only in master process (#89)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-24 17:39:30 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
ae2443c210 Add flops, num_params, inference speed logging and best.pt logging (#84)
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>
2022-12-23 20:14:21 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
40f7e0806e Add v8 modules (#81)
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>
2022-12-21 00:29:58 +01:00
Ayush ChaurasiaGitHubLaughing-qpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
d198f3f9d1 Update docs (#73)
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-20 09:53:55 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
681cfc1c35 Make config overrides user friendly (#80)
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>
2022-12-20 09:50:08 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
48c95ba083 Change class depending on dataset in model interface (#77)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-16 13:56:25 +05:30
Ayush ChaurasiaandGitHub 24a7c068ad update docs (#76) 2022-12-15 11:48:03 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
eb5adf4e0b Model enhancement (#75)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-14 14:33:31 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
d85b44f259 Update docs (#71)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-12 09:21:00 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Laughing-q
7ae45c6cc4 [WIP] Model interface (#68)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
2022-12-08 13:22:56 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Laughing-qGlenn Jocher
e6737f1207 Predictor support (#65)
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>
2022-12-07 10:33:10 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
fe75a9ce67 docs setup (#61)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-05 06:04:57 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Laughing-q
7ec7cf3aef Detection support (#60)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
2022-12-03 21:01:23 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
5a52e7663a standalone val (#56)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-30 15:04:44 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
d0b0fe2592 CLI updates (#58)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-29 15:41:37 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
c5f5b80c04 Deterministic training (#53)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-25 02:06:14 +05:30
ayush chaurasia 793dde365d Revert "update"
This reverts commit 486bc49414.
2022-11-25 01:54:50 +05:30
ayush chaurasia 486bc49414 update 2022-11-25 01:11:55 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
d7df1770fa Add warmup and accumulation (#52)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-25 00:42:16 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
298287298d Add clearml logging (#51)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-24 15:25:03 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
512a225ce8 Cli support (#50)
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>
2022-11-23 11:37:27 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
4291b9c31c Add EMA and model checkpointing (#49)
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>
2022-11-19 19:07:26 +01:00
Ayush ChaurasiaandGitHub db1031a1a9 Allow setting model attributes before training (#45) 2022-11-17 10:35:53 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
832ea56eb4 update model initialization design, supports custom data/num_classes (#44)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-15 15:36:29 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1f3aad86c1 new check_dataset functions (#43)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-12 15:02:41 +01:00
Ayush ChaurasiaandGitHub d143ac666f [rename] - preprocess-batch -> preprocess, preprocess_preds -> postprocess (#42) 2022-11-09 17:35:15 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
f56c9bcc26 Segmentation support & other enchancements (#40)
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>
2022-11-08 16:27:57 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
92c60758dd Smart Model loading (#31)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-31 23:52:12 +01:00
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1054819a59 Add initial model interface (#30)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-25 21:51:15 +02:00
Ayush ChaurasiaGitHubAyush Chaurasiapre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
7b560f7861 Model builder (#29)
Co-authored-by: Ayush Chaurasia <ayush.chuararsia@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-18 23:14:23 +02:00
Ayush ChaurasiaGitHubAyush Chaurasiapre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
c5cb76b356 Metrics and loss structure (#28)
Co-authored-by: Ayush Chaurasia <ayush.chuararsia@gmail.com>
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>
2022-10-15 19:39:05 +02:00
Ayush ChaurasiaGitHubLaughing-qpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn JocherAyush ChaurasiaAyush Chaurasia
d0b3c9812b Trainer + Dataloaders (#27)
Co-authored-by: Laughing-q <1185102784@qq.com>
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 <ayushchaurasia@Ayushs-MacBook-Pro.local>
Co-authored-by: Ayush Chaurasia <ayush.chuararsia@gmail.com>
2022-10-10 14:01:07 +02:00
Ayush ChaurasiaandGitHub 419f6d7167 Add yolo package structure (#15) 2022-09-19 15:47:23 +05:30
Ayush ChaurasiaGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
39e9f3cae9 Initial pip structure (#1)
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>
2022-09-16 22:43:06 +02:00