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()