MindSpore实现手写数字识别代码

最新推荐文章于 2023-06-20 12:29:27 发布

qq_49723795

最新推荐文章于 2023-06-20 12:29:27 发布

阅读量1.9k

点赞数 1

文章标签：深度学习神经网络

本文链接：https://blog.csdn.net/qq_49723795/article/details/120665220

版权

MindSpore是华为自研的一套AI框架，最佳匹配昇腾处理器，最大程度地发挥硬件能力。作为AI入门的LeNet手写字体识别网络，网络大小和数据集都不大，可以在CPU上面进行训练和推理。下面是基于MindSpore的LeNet手写字体识别代码，直接复制到ubuntu的Jupyter即可以运行，但是要确保安装了Mindspore包哦~

MNIST数据集需要提前准备好放在目录中。

import os
import argparse
from mindspore import context

parser = argparse.ArgumentParser(description='MindSpore LeNet Example')
parser.add_argument('--device_target', type=str, default="CPU", choices=['Ascend', 'GPU', 'CPU'])
args = parser.parse_known_args()[0]
context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target)

"""!mkdir -p ./datasets/MNIST_Data/train ./datasets/MNIST_Data/test
!wget -NP ./datasets/MNIST_Data/train https://mindspore-website.obs.myhuaweicloud.com/notebook/datasets/mnist/train-labels-idx1-ubyte --no-check-certificate
!wget -NP ./datasets/MNIST_Data/train https://mindspore-website.obs.myhuaweicloud.com/notebook/datasets/mnist/train-images-idx3-ubyte --no-check-certificate
!wget -NP ./datasets/MNIST_Data/test https://mindspore-website.obs.myhuaweicloud.com/notebook/datasets/mnist/t10k-labels-idx1-ubyte --no-check-certificate
!wget -NP ./datasets/MNIST_Data/test https://mindspore-website.obs.myhuaweicloud.com/notebook/datasets/mnist/t10k-images-