CASIA-IVA-Lab/DPT

Where can I download pretrained/dpt_tiny.pth?

lingblessing opened this issue · 5 comments

Where can I download pretrained/dpt_tiny.pth?

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

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

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)

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