[P] Extract Local Attention Positions

I am working on a project, I want to extract the local attention positions present in the feature maps of every level in the FPN. Below is the visualization:

​

https://preview.redd.it/y6t3koh91bm81.jpg?width=768&format=pjpg&auto=webp&s=7f5eee4229a35ec0c173b0417aa2a6eeb2c7c761

I am using [MMDetection](https://github.com/open-mmlab/mmdetection) toolbox for editing.

For example, a level in the FPN outputs the feature map that can be visualized below:

​

https://preview.redd.it/ypi2fmtd1bm81.jpg?width=374&format=pjpg&auto=webp&s=f1414b1d9cfc41969fb4866bdf9ecc56a659e761

I want am to know how to locate those weights and how to extract their location with respect to the (H x W x …

attention local attention machinelearning