pytorch中如何查看中間特征圖

（1）直接在代碼運行中查看：

self.feature_map = feat.detach().cpu()
                    
for i in range(self.feature_map.size(1)):                           
     plt.matshow(self.feature_map[0,i,:,:],cmap="viridis")
     plt.draw()
     plt.pause(0.1)
     plt.close()    

或者

def viz(input):
    x = input[0]
    print(x.shape)
    min_num = np.minimum(25, x.size()[0])
    for i in range(min_num):
        plt.subplot(5, 5, i+1)
        plt.imshow(x[i].cpu())
    plt.show()


self.feature_map = feat.detach().cpu()
viz(self.feature_map)

 

（2）保存到本地查看

from torchvision.utils import save_image

for i in range(self.feature_map.size(1)):
     save_image(self.feature_map[0][i], os.path.join('../feature_maps', 'image_{}.jpg'.format(i)),
                           nrow=1, padding=0)