Matching mutiple pairs at the same time

[Kudo510]

Hi,
it is shown in the demo how to find mathces between 2 single images. Is there a way that we can do it for batches e.g 4 images to 4 images or even to output matches of 16 pairs (each image of this batch to another image in the other batch)

[wyf2020]

Great question! The current code supports '4 images to 4 images', you can change the hard code here to args.batch_size, and then change the batch_size in the test script to N. If your GPU utilization is not 100% when batch_size=1, increasing the batch_size to reach 100% utilization can significantly improve throughput. The current code does not support outputting '16 pairs', you can modify the current code to only compute backbone four times in batch manner, and perform the attention and subsequent operations 16 times in batch manner.