Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to contribute pretrained models? #2445

Open
veb-101 opened this issue May 15, 2024 · 2 comments
Open

How to contribute pretrained models? #2445

veb-101 opened this issue May 15, 2024 · 2 comments

Comments

@veb-101
Copy link

veb-101 commented May 15, 2024

Hello, I've successfully converted MobileViT v1 (variants - XXS, XS, S) models from Pytorch to Keras 3, including their pre-trained image classification weights. Can such models be added to KerasCV? If so, can you point me to the correct contributing guide?

As part of my commitment to this project, I've already created a separate repo to push other versions of MobileViT and other models.

Please try it out over here before you make any decision: https://colab.research.google.com/github/veb-101/keras-vision/blob/main/examples/mobile_vit_v1.ipynb

Repo: keras-vision

Currently, the weights are hosted in my repository releases.

@sachinprasadhs
Copy link
Collaborator

Thank you for your interest in contribution, we don't have any documentations as of now, but you can refer our existing implementations here to create a PR https://github.com/keras-team/keras-cv/tree/master/keras_cv/src/models/backbones

@veb-101
Copy link
Author

veb-101 commented May 16, 2024

Thanks, I'll work on porting MobileViT v2 models first and then create a PR for both versions.
You may close or keep this issue open.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants