摘要
本节主要是学习keras的相关模型搭建,通过猫狗分类来进一步学习keras
样本数据预处理
- 将数据集中的train压缩包解压
- 得到train文件夹,并将该文件夹中的样本数据分为3部分train,validation,test
- 每部分的样本都包括dog和cat文件夹(都来自train的解压包数据)
训练神经网络
- 三层卷积神经网络
- 2层全连接网络
- 数据增强手段
- 开始模型训练
使用模型
保存模型训练参数
根据前面的训练模型,搭建可识别的网络
随便选择一个test文件夹中的图片
输出,猫和狗的类名
[x] Edit By Porter, 积水成渊,蛟龙生焉。