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

Where can I download pretrained/dpt_tiny.pth? #8

Open
lingblessing opened this issue Dec 21, 2021 · 5 comments
Open

Where can I download pretrained/dpt_tiny.pth? #8

lingblessing opened this issue Dec 21, 2021 · 5 comments

Comments

@lingblessing
Copy link

Where can I download pretrained/dpt_tiny.pth?

@volgachen
Copy link
Collaborator

Please refer classification/README.md

@lingblessing
Copy link
Author

How do I use this plug-and-play DPT?

@lingblessing
Copy link
Author

I want to use the DPT plug-and-play module. Is there a procedure? The kind of departure from the entire PVT code.

@volgachen
Copy link
Collaborator

Yes. If you want to insert the plug-and-play module into other models, please refer these steps:

  1. Make sure you have compared the operation
  2. Copy box_coder.py, depatch_embed.py, ms_deform_attn_func.py to your model directory. (Ensure that they are in the same folder)
  3. Import the modules and use it as in this file
  4. args and kwargs depends on your project.
from .box_coder import *
from .depatch_embed import Simple_DePatch

box_coder = pointwhCoder(*args, **kwargs)
your_module = Simple_DePatch(box_coder, *args, **kwargs)

@lingblessing
Copy link
Author

Yes, I know it is used this way, but my current model is limited by img_size, is there any way I can ignore this variable

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