这是我的可视化代码:
如何在这里使用
f, ax = plt.subplots(1, 2)
for i, img in enumerate([img1, img2]):
grads = # my visualization codes
# visualize grads as heatmap
ax[i].imshow(grads, cmap='jet')
如何在这里使用
imshow
保存显示的内容?非常感谢任何建议!