[問題] numpy維度問題

作者: grimaceo (哈囉哈囉)   2020-05-16 11:17:41
各位前輩好 小的最近在建構CNN時遇到一些問題
當我讀進訓練集N張照片時train.shape會顯示(N,1) train[0][0].shape就會顯示(height
,w
但model要讀資料時會要求資料要是四維的array,網路上好像都找不到能把二維轉成四維

試過resize但資料好像會跑掉 想請問各位:
有甚麼方法可以把二維array中的array降維變成4D矩陣,或是有甚麼方法讓CNN網絡能夠
讀?
謝謝各位
作者: littlebo1990 (Bo)   2020-05-16 11:54:00
train.reshap(-1,3,3,1) 選我正解抱歉搞錯了XD 我以為你w,h分別為3, channel是1..應該是你的np array裡面是object, 所以shape才是你可以用 list comprehension 把裡面的object變arrai.e train = np.array([i.reshape(w,h,3) for i intrain])這樣train dim就會變(N,w,h,3)
作者: eooft   2020-05-16 13:11:00
那個3代表channel數 先弄清楚你的channel是1(黑白)還3(RGB)

Links booklink

Contact Us: admin [ a t ] ucptt.com