基于cnn的图像二分类算法(一)

本算法是基于tensorflow,使用python语言进行的一种图像分类算法,参考于谷歌的mnist手写识别,包括如下几个模块:图像读取,图像处理,图像加强。卷积神经网络部分包括:卷积层1,汇合层1(部分文献也有叫池化层的),卷积层2,汇合层2,全链接层1,全链接层2,共6层神经网络。损失函数采用交叉熵,优化则采用adam优化法,因为数据集大小较小,只有200张图片,故没有采用MBGD梯度降低算法
相关文章
相关标签/搜索