Is it possible to run inference in batch mode

cvg / LightGlue

LightGlue: Local Feature Matching at Light Speed (ICCV 2023)

Apache License 2.0

3.28k stars 313 forks source link

Is it possible to run inference in batch mode #61

Open pumetu opened 1 year ago

pumetu commented 1 year ago

Hello, I was wondering if it would be possible if lightglue could be ran in batch mode, since I have to match a large amount of images the bottleneck right now is inference time. Thank you for your work!

sarlinpe commented 12 months ago

Yes but you'd need to:

disable adaptive depth and width mechanisms.
make sure that the detector finds a fixed number of keypoints in all images, using max_num_keypoints=1024, detection_threshold=0.0

mmmmmjaai commented 11 months ago

hi can you share you code in batch mode with me? iam beginer, i got some trouble with it, thank you

kkamalrajk commented 5 months ago

@sarlinpe Could you please help me with the batch processing code ?

nguyenhoangvudtm23 commented 1 month ago

@sarlinpe what do you think about the performance if I use zero-pad for keypoint and disable point_pruning? May it be significantly reduced?