1094 Commits

Author SHA1 Message Date
Ayush Chaurasia
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
Laughing
3a241e4cea
update segment training (#57)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: ayush chaurasia <ayush.chaurarsia@gmail.com>
2022-11-29 17:00:08 +05:30
Ayush Chaurasia
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 Chaurasia
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 486bc494142744e9d90f0483b611a785381401a7.
2022-11-25 01:54:50 +05:30
ayush chaurasia
486bc49414 update 2022-11-25 01:11:55 +05:30
Ayush Chaurasia
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 Chaurasia
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 Chaurasia
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 Chaurasia
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
Glenn Jocher
27d6545117
General console printout updates (#48)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-19 16:08:16 +01:00
Glenn Jocher
8530e3fae0
Logging fix from YOLOv5 (#47)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-19 14:51:52 +01:00
Laughing
47f1cb3ef4
Fix some cuda training issues of segmentation (#46)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-17 18:14:02 +05:30
Ayush Chaurasia
db1031a1a9
Allow setting model attributes before training (#45) 2022-11-17 10:35:53 +05:30
Ayush Chaurasia
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 Chaurasia
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 Chaurasia
d143ac666f
[rename] - preprocess-batch -> preprocess, preprocess_preds -> postprocess (#42) 2022-11-09 17:35:15 +01:00
Glenn Jocher
4c68b9dcf6
Add is_colab() and is_kaggle() (#41)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-08 17:14:49 +01:00
Ayush Chaurasia
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
Laughing
c617ee1c79
Fix dataloader2 (#35)
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 12:47:25 +01:00
Glenn Jocher
523eff99e2
Update ci.yaml (#37)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-07 19:27:55 +01:00
dependabot[bot]
a733e41bcc
Bump actions/setup-python from 3 to 4 (#39)
Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
2022-11-07 19:11:14 +01:00
Glenn Jocher
06dd49d862
Create dependabot.yml (#38) 2022-11-07 19:04:53 +01:00
Glenn Jocher
e4f7458d90
Hydra *.yml extension deprecated fix (#34) 2022-11-07 00:52:08 +01:00
Glenn Jocher
6fe8bead35
Classify training cleanup (#33)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-07 00:15:57 +01:00
Laughing
2e9b18ce4e
Fix dataloader (#32)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-04 13:30:40 +05:30
Ayush Chaurasia
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 Chaurasia
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 Chaurasia
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 Chaurasia
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 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
Glenn Jocher
7a2e5fdfa3
Update ci.yaml (#21) 2022-09-21 13:57:28 +02:00
Glenn Jocher
abbd177a78
Update README.md 2022-09-21 13:45:12 +02:00
Glenn Jocher
f5dd65c709
Update README.md 2022-09-21 13:42:29 +02:00
Glenn Jocher
2472feafad
Add missing setup.py fields (#18) 2022-09-19 15:49:09 +02:00
Glenn Jocher
171ad46d6b
Create __init__.py (#17)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-09-19 15:19:13 +02:00
Ayush Chaurasia
419f6d7167
Add yolo package structure (#15) 2022-09-19 15:47:23 +05:30
Glenn Jocher
64212b67b9
tensorflow-macos comment (#14) 2022-09-18 17:14:46 +02:00
Glenn Jocher
3c58c26569
Update requirements.txt with fire>=0.4.0 (#12) 2022-09-17 11:41:08 +02:00
Glenn Jocher
088bd5b462
Create ci.yaml (#10)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-09-17 00:50:20 +02:00
Glenn Jocher
4aa74bb726
Normalize version with dev0 (#11) 2022-09-17 00:06:11 +02:00
Glenn Jocher
745f8e3759
Update requirements.txt (#9) 2022-09-16 23:42:08 +02:00
Ayush Chaurasia
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
Glenn Jocher
95f6fa16d5
Initial commit 2022-09-11 19:39:46 +03:00