MindSpore 21天实战营---MindSporeLite_mindspore cifar-10-binary

作者：从前慢现在也慢 | 2024-03-12 05:27:20

踩

mindspore cifar-10-binary

体验作业APP下载包：

https://mslite-app.obs.cn-north-4.myhuaweicloud.com/app-ms.apk?AccessKeyId=PQ7DQUATQUMX3VMMPIPM&Expires=1606143317&Signature=Of6H2O9/tVpZx6KyIepv8/la//g=

下载完成后直接在安卓手机上进行安装即可使用

运行效果图：

进阶作业

模型训练过程：

下载mobilenetv2体验脚本

1234	# 在root用户主目录下执行如下命令git clone https://github.com/mindspore-ai/mindspore-21-days-tutorials.gitmkdir -p /root/workspace/mobilecp -r /root/mindspore-21-days-tutorials/chapter1/mobilenetv2 /root/workspace/mobile

准备cifar-10数据集（binary二进制格式）

将下载好的数据集解压，生成5个训练集.bin文件和1个测试集.bin文件。

123456789

# 下载并解压cifar-10数据集, 生成5个训练集.bin文件和1个测试集.bin文件wget http://www.cs.toronto.edu/~kriz/cifar-10-binary.tar.gztar -zxvf cifar-10-binary.tar.gz# 创建用于存放训练集的目录，将训练集5个.bin文件拷贝到该目录下mkdir -p /root/workspace/mobile/data/traincp /root/cifar-10-batches-bin/data_*.bin /root/workspace/mobile/data/train# 创建用于存放测试集的目录，将测试集1个.bin文件拷贝到该目录下mkdir -p /root/workspace/mobile/data/cp /root/cifar-10-batches-bin/test_batch.bin /root/workspace/mobile/data/

训练启动阶段

启动GPU容器

使用GPU mindspore-1.0.0版本镜像，将训练脚本及数据集所在目录挂载到容器环境中

12	docker run -it -v /root/workspace/mobile:/mobile --runtime=nvidia --privileged=true mindspore/mindspore-gpu:1.0.0 /bin/bash开始训练

12	cd /mobile/mobilenetv2python train.py --is_training=True --epoch_size=10

验证结果

1	python .py --is_training=False --pretrain_ckpt=ckpt_0/mobilenetv2-10_1562.ckpt

注：需事先安装好docker gpu环境，docker可以通过apt-get install docker进行安装（ubuntu系统）

进阶作业打包APK操作可以参考码云教程：https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/lite/object_detection

转自文章链接：MindSpore 21天实战营-MindSporeLite_MindSpore_昇腾论坛_华为云论坛

感谢作者的努力与分享，侵权立删！

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/222367