使用Tensorflow搭建CNN網絡處理MNIST

我們這次搭建的神經網絡如下圖所示: 它的輸入是一個像素值爲28*28的灰度圖,然後輸入的數據先經過一個卷積層,卷積核的大小是5*5*32,得到32個feature map,之後經過池化層,這裏我們選擇了最大池化,得到12*12*32的數據,此時再經過一個卷積層,卷積核爲5*5*64,得到的結果再一次經過池化,得到的數據爲4*4*64,最後通過一個全連接層,得到最終的結果。 1.下載和準備數據 在這
相關文章
相關標籤/搜索