1. 程式人生 > >python/numpy/ndarry陣列擴維

python/numpy/ndarry陣列擴維

二維陣列擴充套件為三維陣列(此方法增加了第一維,後同)

a=[[1,2],[2,3]]
b=[[4,5],[6,7]]
c=np.array([a,b])
print(c)

從圖片資料夾下讀取圖片,實現ndarry儲存

假設已讀取n個彩色影象,儲存在img[n]中,(高*寬*通道數)

data=np.zeros(n,35,43,3])              #開闢四維陣列

for i in range(0,n)

   img1=transform.resize(img[i],(35,43))    #將所有圖片變換成同樣大小(高*寬),這裡變換為(35*43)

   data[i]=img1

這樣獲得的data就是BHWC(Batchsize, Height, Width, Channel)形式了