MindSpore v1.0实现简单的图片分类应用

因为时间不是很多，就使用aconda创建配置了相关环境，安装了mindspore v1.0版本，具体使用可查阅相关资料，安装没有问题。

image.png

选择命令行安装，如上图所示，安装在相应目录下

image.png

下载数据集，前两个训练用，后两个测试用，下载地址官网教程有。

image.png

查看数据集，是否正常

image.png

符合要求，我们直接开始数据处理，数据处理代码参考教程。

处理完毕后我们查看处理后的数据

image.png

image.png

通过上述查询操作，看到经过变换后的图片，数据集内分成了1875组数据，每组数据种含有32张图片，每张图片i像素位32*32，数据全部准备号后，就可以开始下一步了

构造神经网络

image.png

接下来是自定义回调函数手机模型的损失值和精度值，搭建网络并进行训练

定义损失函数优化器

image.png

image.png

最后测试模型精度

image.png

image.png

从上图可看出精度达到了96%，模型优良，符合预期。

 

感谢，祝mindspore越来越好

posted @ 2021-12-31 14:49  MS小白  阅读(29)  评论(0)    收藏  举报