【摘要】本次实践基于mobilenetV2 实现猫狗图像分类,贯穿了数据集获取及处理、预训练模型微调及迁移、端侧部署及推理等环节和知识点,体会到了MindSpore 简单的开发体验和全场景快速部署的魅力。
作者:胡琦
startTime: 2021年1月23日00:43:22
endTime: 2021年1月23日11:34:44
(包含学习、睡觉、吃饭、爬坑、水文……的时间)
了解MindSpore开源生态发现一个小秘密。 github 上多三个仓库,是什么呢?
此处应是 github 过滤的 bug ,实际上mindspore-ai 有 15 个仓库, 比 gitee 多出来的三个是:
mindspore-21-days-tutorials、 mail_templates、 infrastructure
别看这 3 个不起眼的仓库,在社区建设方面却大有作为。比如,mindspore-21-days-tutorials 是我们之前参加 21 天实战营的参考代码和指导文档,多么宝贵的学习资料;另外两个是 MindSpore 的开源基础建设,其中 infrastructure 包含了用于配置 Mindspore 社区的所有必需 Dockerfile 和 YAML 文件,并借助 Github 的 Action 定时自动同步 Gitee 的代码到 Github 。
训练时长截图手机识别截图
学结
在学习之前我以为我会了,但真正实践起来还是磕磕碰碰,所有的代码似曾相识,都是 26 个字母加一些符号组成,但真正去理解还是发现基本功不够:一是不知道怎么写,二是不知道为什么要这么写。尽管如此,我大概理顺了整个实践流程:训练猫狗图像分类模型(云端) --> 手机端推理及应用 --> 从 “1” 开始 Fine Tune 模型(本地) --> 手机端验证 ,当然作为学渣,整个操作过程肯定不止一个小时, 深刻体会到“眼睛:学会了,脑子:学废了!”,因此必须借此帖记录一下“学废了”的过程。
目的本次实践基于 mobilenetV2 实现猫狗图像分类,贯穿了数据集获取及处理、预训练模型微调及迁移、端侧部署及推理等环节和知识点,体会到了 MindSpore 简单的开发体验和全场景快速部署的魅力。
项目目录依赖安装
本次实践依赖opencv-python和 matplotlib,一个用来处理图形比如打印图片和嵌入文字,一个用来将数据集以可视化图片的形式展现出来。
数据预处理
主要删除一些不符合要求(如非JPEG格式)的图片并分割 train 和 eval 数据集,默认9:1:
Fine tune
执行Fine tune脚本train.py,并生成模型文件:
此时会弹窗提示6张图片,这是因为脚本会在正式训练前从数据集中抽取6张图片载入当前模型文件,需要手动关闭才能继续,这是 AI 很傻,全都识别成了 Dog 。
当然,我这边由于机器的原因,没少踩坑,这不“出师未捷身先死”,突然的报错把我整蒙了,将所有
num_parallel_workers
参数设置为4之后,继续训练!
这一步耗时就和本地机器的性能有关了,我的粗粮渣渣机大概跑废了。趁着训练的空档,学习了一些代码。作为新晋调参学徒,我大概知道参数和配置在code\src\args.py和code\src\config.py这两个文件,而数据的加载在code\src\dataset.py这个文件处理,code\src\mobilenetV2.py定义了模型,code\src\models.py这个文件读取和保存模型并打印输出训练日志。
训练完成会,又会调用predict_from_net方法来显示预测的图片和标签,这回我们发现 AI 挺棒的,全部识别正确! 当我满怀信心点击关闭的时候,以为程序立马会给我一个mobilenetv2.mindir,结果我又蒙了,返回了一串错误日志!!!
一开始我以为是文件层级太深了,将文件目录迁移到盘的根目录,重来重来!好在不需要再次加载数据集了,不然又得漫长的等待。 满怀信心结果又被“啪啪啪”打脸,之后群里请教一遍之后,王辉老师建议我把路径改成文件名再试试,果然立马奏效~
终于如愿以偿,code\mobilenetv2.mindir她来了!
手机端推理及应用
训练模型转换
将.mindir模型文件转换成.ms文件,.ms文件可以导入端侧设备并基于MindSpore端侧框架训练。
F:\MindSporePetClassification\converter>call converter_lite --
fmk=MINDIR --
modelFile=f:\MindSporePetClassification\code\mobilenetv2.mindir --outputFile=pet
我们可以下载 MindSpore 官方提供的 AndroidAPP 源码:
https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/lite/pet_classification
或者直接下载打包好的 APP 安装到手机:
https://download.mindspore.cn/model_zoo/official/lite/apk/pet/petclassification.apk
先体验下预训练模型的识别效果:
接着我们把转换好的模型移动到手机端的/sdcard/PetClassification,这里用到的是 ADB 工具:需要确保手机已开启开发者模式并打开文件传输
再试试识别效果:
对本次实践 APP 端代码感兴趣的小伙伴可以直接去阅读源码:
https://gitee.com/mindspore/mindspore/blob/master/model_zoo/official/lite/pet_classification/app/src/main/java/com/mindspore/classificationforpet/widget/MainActivity.java
本文分享自华为云社区《Copy攻城狮1小时入门AI开发工程师》,原文作者:胡琦。
点击关注,第一时间了解华为云新鲜技术~
ai程序如何编写(仅需CtrlC和V一小时带你入门AI开发工程师)
2024-11-21 09:55:30 作者:你大颖姐 网址:https://m.xinb2b.cn/tech/nfc129517.html
- 每天学习生活的汉语句子(每天积累一些好句子121)
- 2024-11-22每天积累一些好句子1211、没有人愿意听到坏消息,所以人们往往低估坏事发生的概率——《大空头》2、当我最想说些什么的时候,往往也是我最沉默的时候——《萌芽》3、不解释就明白不了的东西,即便解释了也明白不了——村上春树4、低头。
- 八年级物理作图题含答案(八年级物理上册所有类型)
- 2024-11-22八年级物理上册所有类型今天老师给大家准备了八年级物理上册所有类型“作图题”汇总,需要打印版的家长可以翻到文末看领取方式哦~如果还想要其它年级或者学科资料的家长们,可以在下方评论区留言或者私信老师索取哦~一、成像知识1.成像。
- 历史上连中三元的人物多少个(看一看历史上连中三元的牛人)
- 2024-11-22看一看历史上连中三元的牛人从隋唐开始,我国就实行科举取士,用考试来选拔人才读书人要想从政做官,必须通过皇帝举行的科举考试所谓朝为田舍郎,暮登天子堂,科举成了天下读书人改变命运和实行理想的唯一通道,也被民间戏称为“跃龙门”在科举。
- 如何电脑连电视(电脑接电视步骤)
- 2024-11-22电脑接电视步骤1.将数据线(VGA或HDMI)的一端插入计算机,另一端插入电视的相应连接2.打开台式计算机或笔记本电脑,然后打开电视3.用电视遥控器选择的信号源是VGA或HDMI(根据数据线选择)4.在台式机或笔记。
- 孔子学而时习之不亦说乎(诸子老子知者不博)
- 2024-11-22诸子老子知者不博“知者不博,博者不知”出自《老子》第八十一章意思是有智慧的人知识不广博,知识广博的人没有智慧虽然我们都知道知识跟智慧是两回事,但是二者真的如此水火不容吗?老子说这句话的逻辑在哪里呢?其实这句话的关键在。
- 家庭装修后甲醛检测标准 房子装完别急着入住
- 2024-11-22家庭装修后甲醛检测标准 房子装完别急着入住房子装完别急着入住,了解室内污染等级,3种甲醛测试方法来帮忙装修污染一直都是很受关注的,这个污染除了建材家具释放出来的之外,还会使用到各种粘合胶水,这些粘合胶水也可能会释放出甲醛等有害气体的在装修之后。
- 推动道路交通安全走深走实(铭记嘱托忠诚担当)
- 2024-11-22铭记嘱托忠诚担当为贯彻落实道路事故预防“减量控大”工作目标,有效减少和遏制涉及货车的交通事故,从源头上消除安全隐患,近日,平利县公安局交通管理大队法宣股联合城区中队,组织平利县昌杰、顺达运输有限公司全体驾驶员开展道路。
- 漫威漫画在哪看(看漫威电影不如看漫威漫画)
- 2024-11-22看漫威电影不如看漫威漫画对于喜爱超级英雄电影的大家来说,好莱坞漫改超级英雄大片算的上是让我们年年进电影院的理由了上周五,作为漫威电影宇宙第三阶段承上启下之作的《惊奇队长》也在国内上映了,「惊奇队长」卡罗尔也凭借在这部电影中的。
- 推文遇到野人老公(被隔离在前男友家中by马蹄糕)
- 2024-11-22被隔离在前男友家中by马蹄糕书名:被隔离在前男友家中作者:马蹄糕简介甘甜将江以恒评为“最烂前任”厌恶到分手搬家时,连泰迪熊都忘抱走就在她重返K市去他家取熊的瞬间,小区惨遭封闭-网上有段子手说隔离期间,一男一女不是造孩子就是闹离婚。
- 世界航空母舰最大排水量多少吨(世界上最大油轮排水量82万吨)
- 2024-11-22世界上最大油轮排水量82万吨说起世界上最大的船,可能不少人都会首先想到航空母舰是啊,这种高端军用舰艇动辄拥有数万吨的满载排水量,加上霸气的外形,往往会给人留下十分深刻的印象但其实,航空母舰在巨型油轮面前,还是显得有些娇小世界上最。
- 无花果辅食炖汤(孕妇怎么吃无花果)
- 2024-11-22孕妇怎么吃无花果无花果含有多种人体必需氨基酸、有机酸、维生素多种营养物质,对健康高的帮助很大孕妇也是可以食用的那孕妇怎么吃无花果好呢?下面来看看无花果煲汤的做法无花果是一种营养丰富的水果,食用方法非常高的多新鲜的无花。
- 七月的风 八月的雨是什么歌
- 2024-11-22七月的风 八月的雨是什么歌《遥远的你》歌词眼角闪起的泪光,无边无际地游荡眉下的一记荒唐,庸俗地写在脸上犹如一缕阳光,但却觉得感伤记忆挥散不去,却藏进了我的诗句,七月的风八月的雨,卑微的我喜欢遥远的你你还未来,怎敢老去未来的我和。