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

Vit model weights #54

Open
gautham-98 opened this issue May 16, 2024 · 0 comments
Open

Vit model weights #54

gautham-98 opened this issue May 16, 2024 · 0 comments

Comments

@gautham-98
Copy link

  1. Why does build_prototypes.ipynb use model = torch.hub.load('facebookresearch/dinov2', 'dinov2_vitl14') while inference in demo.py uses model_path="weights/trained/open-vocabulary/lvis/vitl_0069999.pth"? Should'nt the models be the same ?

  2. Why do we have different weights file for COCO and LVIS? as per my understanding the Vit model remains the same which is DinoV2 even if the datasets changes, since there is no fine tuning.

  3. Finally for a custom dataset if there is no domain gap the steps would be to 1. create prototypes, 2. Run the demo, and if there is a domain gap i guess we have to fine tune the RPN and leave the Vits part as it is. Would this be the way?

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

1 participant