基於cnn的圖像二分類算法(一)

本算法是基於tensorflow,使用python語言進行的一種圖像分類算法,參考於谷歌的mnist手寫識別,包括如下幾個模塊:圖像讀取,圖像處理,圖像加強。卷積神經網絡部分包括:卷積層1,匯合層1(部分文獻也有叫池化層的),卷積層2,匯合層2,全鏈接層1,全鏈接層2,共6層神經網絡。損失函數採用交叉熵,優化則採用adam優化法,因爲數據集大小較小,只有200張圖片,故沒有采用MBGD梯度降低算法
相關文章
相關標籤/搜索