about visualization on attention maps

[amiltonwong]

Hi, @XiaLiPKU ,

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

THX!

[Xia Li 李夏]

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

[zhudahui]

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