search

Zj-BinXia / ENLCA

This project is official implementation of 'Efficient Non-Local Contrastive Attention for Image Super-Resolution', AAAI2022
86 stars 7 forks source link

Correlation map visualization #7

Closed nanmehta closed 2 years ago

nanmehta commented 2 years ago

Can you please help me out with how we can find the correlation maps? I am not getting it.

Zj-BinXia commented 2 years ago

The softmax(QK) is an HW x HW feature map, we select a point in the first HW (the red point in the figure) and can draw a HW heatmap about the point.