AI常用框架和工具丨8. Keras实现基于CNN的手写数字识别

Posted 2022-05-11 AXYZdong

文章目录

• • 环境说明
  • 一、模型训练
  • • 1.1 导入相关依赖
    • 1.2 数据集导入并编码
    • 1.3 CNN网络搭建
    • 1.4 进行训练
    • 1.5 模型保存
    • 1.6 完整代码
  • 二、本地手写数字批量识别
  • • 2.1 导入相关依赖
    • 2.2 数据和模型准备
    • 2.3 批量预测并显示
    • 2.4 完整代码
    • 2.5 运行结果

环境说明

操作系统：Windows 10

CUDA 版本为: 10.0

cudnn 版本为: 7.6.5

Python 版本为：Python 3.6.13

tensorflow-gpu 版本为：1.13.1