Using VIT as a feature extractor #1518

Unanswered

Bk073 asked this question in General

Bk073
Oct 24, 2022

I want to use VIT as a feature extractor. Suppose I have passed an image of size (3, 224, 224), how can I get output of same size?
How can I remove the final layer which for now return a vector of shape (1000) ?

Replies: 0 comments

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