python之显示matplotlib图片，图片维度处理

1.numpy中升维度a.expand_dims(a,axis=0)

注意unsqueeze(x,axis=0)是torch中的用法，np没有这个属性

 

2.堆叠stack

np.stack((a,b),axis=0)

axis=0表示在批量上堆叠，如果图片维度是(1,2,6,3),那么堆叠之后会是(2,1,2,6,3)

3.所以在有批量的时候的图片应该使用np.connect()

3.matpltlib的显示是(H,W,C),并且是RGB顺序

for feature_map in out_put:
    #[N,C,H,W]->[C,H,W]
    im=np.squeeze(feature_map.detach().numpy())
    #[C,H,W]->[H,W,C]
    im=np.transpose(im,[1,2,0])

    #show top 12 feature maps
    plt.figure()
    for i in range(12):
        ax=plt.subplot(3,4,i+1)
        #[H，W，C]
        plt.imshow(im[:,:,i],cmap='gray')
    plt.show()