search

zju3dv / pats

Code for "PATS: Patch Area Transportation with Subdivision for Local Feature Matching", CVPR 2023
Other
90 stars 7 forks source link

Is it possible to set batch size as > 1? #6

Closed GrayMask closed 7 months ago

GrayMask commented 7 months ago

Thanks for your great work. It performs pretty well in my own dataset! I want to know if it is possible to set batch size as a num > 1, (so that I can input multiple image pairs into model to save time) I notice that the batch size is always set as 1 in your demo.

xuanlanxingkongxia commented 7 months ago

Yes， you can set batch size num >1, however, the coordinates of matches are set as one array, this part of code should be modified when use higher batch size.

GrayMask commented 7 months ago

I see. Thanks for your reply!