TensorFlow猫狗图像识别卷积神经网络
卷积神经网络的猫狗识别项目挺适合初学深度学习的你练手用的,代码基于TensorFlow写的,结构清晰,运行环境建议是Linux。
用的是比较标准的CNN 架构,像卷积层、池化层这些都有,代码里也做了数据增强,比如随机翻转、裁剪、归一化啥的,模型泛化能力还不错。
训练流程也比较标准,分成训练和测试两个模块,train.py
负责建模和训练,eval.py
搞评估,路径配置要注意自己改一下。优化器用的是Adam
,损失函数是交叉熵
,基本都是实用派配置。
哦对了,模型构建用的是tf.keras.Sequential
,你要想扩展也挺方便的,逻辑清晰,适合二次开发。训练集、验证集、测试集都有单独分好,跑完一遍基本就能看出效果了。
如果你正好在找猫狗图像识别的完整项目,或者想练习 CNN 和 TensorFlow,建议你直接下一个看看。
下载地址
用户评论