How to draw the feature maps in Figure 2

[wonlp]

hello, I want to draw a middle feature map same as the Figure 2. Could you tell me how to draw it? Thank you!

[c-yn]

Hi, given a feature map visual, the snippet is as follows:

plt.figure(dpi=300) visual = torch.mean(visual, dim=1) plt.imshow(visual.squeeze(0).detach().cpu().numpy()) plt.axis('off') plt.show()

[wonlp]

ok, I got it. Thank you.