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>
|
2 years ago |
Glenn Jocher
|
27d6545117
|
General console printout updates (#48)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Glenn Jocher
|
8530e3fae0
|
Logging fix from YOLOv5 (#47)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
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>
|
2 years ago |
|
Ayush Chaurasia
|
db1031a1a9
|
Allow setting model attributes before training (#45)
|
2 years ago |
|
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>
|
2 years ago |
|
Ayush Chaurasia
|
1f3aad86c1
|
new check_dataset functions (#43)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Ayush Chaurasia
|
d143ac666f
|
[rename] - preprocess-batch -> preprocess, preprocess_preds -> postprocess (#42)
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
Glenn Jocher
|
523eff99e2
|
Update ci.yaml (#37)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
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>
|
2 years ago |
|
Glenn Jocher
|
06dd49d862
|
Create dependabot.yml (#38)
|
2 years ago |
|
Glenn Jocher
|
e4f7458d90
|
Hydra *.yml extension deprecated fix (#34)
|
2 years ago |
|
Glenn Jocher
|
6fe8bead35
|
Classify training cleanup (#33)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Laughing
|
2e9b18ce4e
|
Fix dataloader (#32)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Ayush Chaurasia
|
92c60758dd
|
Smart Model loading (#31)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Ayush Chaurasia
|
1054819a59
|
Add initial model interface (#30)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
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>
|
2 years ago |
|
Glenn Jocher
|
7a2e5fdfa3
|
Update ci.yaml (#21)
|
2 years ago |
|
Glenn Jocher
|
abbd177a78
|
Update README.md
|
2 years ago |
|
Glenn Jocher
|
f5dd65c709
|
Update README.md
|
2 years ago |
|
Glenn Jocher
|
2472feafad
|
Add missing setup.py fields (#18)
|
2 years ago |
|
Glenn Jocher
|
171ad46d6b
|
Create __init__.py (#17)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Ayush Chaurasia
|
419f6d7167
|
Add yolo package structure (#15)
|
2 years ago |
|
Glenn Jocher
|
64212b67b9
|
tensorflow-macos comment (#14)
|
2 years ago |
|
Glenn Jocher
|
3c58c26569
|
Update requirements.txt with `fire>=0.4.0` (#12)
|
2 years ago |
|
Glenn Jocher
|
088bd5b462
|
Create ci.yaml (#10)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Glenn Jocher
|
4aa74bb726
|
Normalize version with dev0 (#11)
|
2 years ago |
|
Glenn Jocher
|
745f8e3759
|
Update requirements.txt (#9)
|
2 years ago |
|
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>
|
2 years ago |
|
Glenn Jocher
|
95f6fa16d5
|
Initial commit
|
2 years ago |
