search

XiaLiPKU / EMANet

The code for Expectation-Maximization Attention Networks for Semantic Segmentation (ICCV'2019 Oral)
https://xialipku.github.io/publication/expectation-maximization-attention-networks-for-semantic-segmentation/
GNU General Public License v3.0
680 stars 130 forks source link

about visualization on attention maps #17

Closed amiltonwong closed 4 years ago

amiltonwong commented 4 years ago

Hi, @XiaLiPKU ,

Could you also provide the code for visualizing on attention maps, like the responses Z in Fig.5?

THX!

XiaLiPKU commented 4 years ago

cv2.imwrite(z[0, 0].cpu().numpy().reshape(h, w))

zhudahui commented 3 years ago

@amiltonwong Did you succeed？ valueError: cannot reshape array of size 64 into shapec(43,26) thanks