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

Posted 2022-07-20 AXYZdong

代码实例，PyTorch实现基于CNN的手写数字识别，希望对您有所帮助。

文章目录

• • 环境说明
  • 一、模型训练
  • • 1.1 导入相关依赖
    • 1.2 选择使用的硬件
    • 1.3 超参数配置
    • 1.4 准备训练集和测试集
    • 1.5 模型创建
    • 1.6 模型评估指标
    • 1.7 模型训练
    • 1.8 模型测试
    • 1.9 模型保存
    • 1.10 完整代码
  • 二、本地手写数字识别
  • • 2.1 导入相关依赖
    • 2.2 硬件选择
    • 2.3 本地模型加载
    • 2.4 本地图片处理
    • 2.5 图像预测
    • 2.6 完整代码

环