Skip to content

Adding FastViT & MobileViT image classification models #2231

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

Open
veb-101 opened this issue Apr 25, 2025 · 1 comment
Open

Adding FastViT & MobileViT image classification models #2231

veb-101 opened this issue Apr 25, 2025 · 1 comment
Assignees

Comments

@veb-101
Copy link

veb-101 commented Apr 25, 2025

Hi, I've recently finished implementing Apple's FastVit , MobileVit, and MobileViT-v2 image classification models in my repo. I've also ported their weights from the original models. My only pending task for these is making them channel-first compatible, but that's simple. I'm also planning to least train the FastViT models on Pascal VOC and make it available, as there is a lack of model options in TensorFlow for object detection since the closing of TFOD.

Could these models be added to Keras-Hub as well? I couldn't find a Vision-specific contributing guide, or do I need to follow either resnet or mobilenet structure?

My final question is whether these models are needed here? If you guys need support for implementing some other models, you can let me know.

@veb-101
Copy link
Author

veb-101 commented Apr 25, 2025

One more thing, the licences on these models is this: github link
Will it be a problem?

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

No branches or pull requests

2 participants