人工智能:CNN训练MNIST 42000张图片

本次训练框架为keras,后续会有TensorFlow框架的训练过程,请等待更新。

训练数据40600张图片

这里写图片描述

验证数据1400张图片
这里写图片描述

数据准备:

这里写图片描述

训练代码:
这里写图片描述

这里写图片描述

这里写图片描述

训练得分达到 0.9821428568022592
这里写图片描述

验证模型实际效果

这里写图片描述

网上找的图片
这里写图片描述

lena1可以轻松识别出是数字2
这里写图片描述

lena2~lena6 无法识别 lena7~lena10可以识别 看来需要把图片处理为黑底白字 大小28X28

补充说明:当把激活函数设置为relu 模型无法收敛。训练学习率调小为0.002,模型可以收敛,但是存在过拟合。

猜你喜欢

转载自blog.csdn.net/xdy1120/article/details/81289338