search

whai362 / PVT

Official implementation of PVT series
Apache License 2.0
1.73k stars 246 forks source link

The input size problem #30

Closed water-wave closed 3 years ago

water-wave commented 3 years ago

Thank you for your great work.The size of my picture is (256,832),how should I deal with it?Please tell me more details.thanks.thanks

whai362 commented 3 years ago

I would use interpolation to adjust the size of pos_embed. See https://github.com/whai362/PVT/blob/main/detection/pvt.py#L228