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

Will it be a good Idea to add a tutorial to train a Image Classification suing ViT using CLS token. #1862

Closed
sineeli opened this issue May 13, 2024 · 2 comments

Comments

@sineeli
Copy link
Collaborator

sineeli commented May 13, 2024

Short Description

Add possible tutorial on training basic architecture for Vision Transformer Using CLS token.

Also will it be a good idea to add ViT as a possible backbone, I can see keras already has ViT-DEIT and I am interested to contribute on writing the backbone.

Thanks

@sineeli sineeli changed the title Would be a good Idea to add a tutorial to train a Image Classification suing ViT using CLS token. Will it be Would be a good Idea to add a tutorial to train a Image Classification suing ViT using CLS token. May 13, 2024
@sachinprasadhs sachinprasadhs changed the title Will it be Would be a good Idea to add a tutorial to train a Image Classification suing ViT using CLS token. Will it be a good Idea to add a tutorial to train a Image Classification suing ViT using CLS token. May 14, 2024
@sachinprasadhs sachinprasadhs transferred this issue from keras-team/keras-cv May 14, 2024
@sampathweb
Copy link
Contributor

@sineeli - We already have a ViT DET Backbone in KerasCV. https://github.com/keras-team/keras-cv/blob/master/keras_cv/src/models/backbones/vit_det/vit_det_backbone.py#L35

Does this address what you are looking for?

@sineeli
Copy link
Collaborator Author

sineeli commented May 16, 2024

@sampathweb,

Yes the ViT DET exists but wanted to know if we can still include the other variation as well (ViT). If the same backbone solves the purpose then we can ignore ViT.

Thanks for letting me know.

@sineeli sineeli closed this as completed May 16, 2024
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

3 participants