這篇文章主要介紹Python怎么實現CNN的多通道輸入,文中介紹的非常詳細,具有一定的參考價值,感興趣的小伙伴們一定要看完!
CNN可以同時進行多通道的輸入,例如一張彩色圖片可以分解成RGB三個通道輸入給CNN,當使用自己的數據集時,可以通過numpy來實現數據的多通道輸入。
假設我們有兩個組數據a和b:
a = np.linspace(1,100,100) b = np.linsapce(-1,-100,100)
然后將a和b轉變成四維數組,TensorFlow接收的數據時四維數組
a = a.reshape(4,1,5,5) b = b.reshape(4,1,5,5)
這樣我們就得到了兩個batch_size = 4, channel = 1, width = 5, high = 5的四維數組,然后
a = a.transpose(1,0,2,3) b = b.transpose(1,0,2,3)
將a和b的維度參數變成[1,4,5,5],然后使用np.vstack()函數將兩個數組在通道數上疊加
c = np.vstack((a,b))
最后再次使用.transpose()函數將疊加后的四維數組轉換為TensorFlow接受的四維數組
c = c.transpose(1,2,3,0)
這時c就變成了[4,5,5,2],即batch_size = 4, width = 5, high = 5, channel = 2的四維數組。
以上是“Python怎么實現CNN的多通道輸入”這篇文章的所有內容,感謝各位的閱讀!希望分享的內容對大家有幫助,更多相關知識,歡迎關注億速云行業資訊頻道!
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。