CNN可以同時進行多通道的輸入，例如一張彩色圖片可以分解成RGB三個通道輸入給CNN，當使用自己的數據集時，可以通過numpy來實現數據的多通道輸入。
假設我們有兩個組數據a和b：
|
1
2
|
a = np.linspace(1,100,100)b = np.linsapce(-1,-100,100) |
然后將a和b轉變成四維數組，TensorFlow接收的數據時四維數組
|
1
2
|
a = a.reshape(4,1,5,5)b = b.reshape(4,1,5,5) |
這樣我們就得到了兩個batch_size = 4, channel = 1, width = 5, high = 5的四維數組，然后
|
1
2
|
a = a.transpose(1,0,2,3)b = b.transpose(1,0,2,3) |
將a和b的維度參數變成[1,4,5,5]，然后使用np.vstack()函數將兩個數組在通道數上疊加
|
1
|
c = np.vstack((a,b)) |
最后再次使用.transpose()函數將疊加后的四維數組轉換為TensorFlow接受的四維數組
|
1
|
c = c.transpose(1,2,3,0) |
這時c就變成了[4,5,5,2]，即batch_size = 4, width = 5, high = 5, channel = 2的四維數組。
以上這篇Python實現CNN的多通道輸入實例就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持服務器之家。
原文鏈接：https://blog.csdn.net/LraoAspire/article/details/89670032
