Different query positions on the same image

[horanyinora]

Hello,
Thanks for the excellent work. I would like to show different attention maps produced using different query positions (reference bouding boxes).
For example when there are three bouding boxes on the image I would like to calculate the attention w.r.t. them and show the difference. Is it possible to do this?

[cmsflash]

It is possible. You can simply multiply a specific query to the key feature map to get the query-specific weights for the entire image.