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.

Sign up for GitHub

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

Jump to bottom

acquire patch tokens #396

Closed

wangyf8848 opened this issue Mar 18, 2024 · 1 comment

wangyf8848 commented Mar 18, 2024

Hello, may I ask if I can get the code of the model? I want to get the patch tokens or even the output of the middle layer, not just the class token.

Contributor

qasfb commented May 13, 2024

The get_intermediate_layers function is what you need !
The model code is in https://github.com/facebookresearch/dinov2/blob/main/dinov2/models/vision_transformer.py

qasfb closed this as completed

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