基于华为自研MindSpore深度学习框架构建网络模型,实现MNIST手写体识别实验。
包含可运行源码、运行结果演示视频,本地MindSpore详细配置教程(私信可远程配置)
本例子会使用MindSpore深度学习框架实现一个简单的图片分类实验,整体流程如下:
1、 处理需要的数据集,这里使用了MNIST数据集。
2、 定义一个网络,这里我们使用LeNet网络。
3、 定义损失函数和优化器。
4、 加载数据集并进行训练,训练完成后,查看结果及保存模型文件。
5、 加载保存的模型,进行推理。
6、 验证模型,加载测试数据集和训练后的模型,验证结果精度。