Skip to content

Latest commit

 

History

History
11 lines (10 loc) · 695 Bytes

MNIST.md

File metadata and controls

11 lines (10 loc) · 695 Bytes

项目描述

在OpenCV中,利用鼠标绘图,然后将绘制的图片送入TensorFlow训练好的模型中,得到训练结果,当用户按下a的时候,撤销刚刚所绘制的图像并显示预测的结果。

简单的实现步骤


  1. 导入数据,进行归一化处理
  2. 构造模型,定义损失函数,精度函数,优化函数
  3. 进行训练,保存模型
  4. 加载模型,定义绘图函数
  5. 将绘制的图片送入模型中预测,判断用户是否绘制完成,如果完成则撤销刚刚绘制的图像,并打印预测类别
  6. 训练好的模型可以在当前文件夹下下载,数据集挺大的上传不了,可以去kaggle上下载