YOLOv3初体验

举报
ModelArts upupup 发表于 2020/05/30 17:19:46 2020/05/30
【摘要】 一、算法订阅:首先登录华为云ModelArts的AI市场,随后选择左侧栏目中的云端算法,最后选择YOLOv3 ResNet18网络进行订阅,因为限时免费可以直接订阅。二、准备数据集可以使用OBS Browser将自己需要训练的数据集传入一个空路径桶这里需要注意目标检测的数据集是类似VOC数据集的格式,即一张图片文件一张xml文件,OBS Browser下载地址:https://support...

一、算法订阅:

首先登录华为云ModelArtsAI市场,

image.png

随后选择左侧栏目中的云端算法,

image.png

最后选择YOLOv3 ResNet18网络进行订阅,因为限时免费可以直接订阅。

image.png

二、准备数据集

可以使用OBS Browser将自己需要训练的数据集传入一个空路径桶

这里需要注意目标检测的数据集是类似VOC数据集的格式,即一张图片文件一张xml文件,

image.png

OBS Browser下载地址:https://support.huaweicloud.com/browsertg-obs/obs_03_1003.html

传完要训练的数据集后在ModelArts主页选择创建数据集,

image.png

image.png

在这里数据集的输入位置选择刚刚上传的路径,输出选择一个空目录即可,记得标注场景选择物体检测。

这里还有一点需要提醒一下,这个算法最后数据集做不做数据切分关系到的是会不会做边训练边验证,所以尽可能做一点切分这样能直观感受到这个算法的精度。

三、训练

训练的话找到算法管理里面的市场订阅,再同步之后直接创建训练作业即可,在这里数据集选择我们刚刚制作的数据集路径,输出选择一个空目录,

image.png

调优的参数的话可以参考算法描述中的文档内容,我这里选择了一个口罩数据集进行训练

image.png

OK,保持默认参数选择资源池后,让我们训练一把!

经过大概10分钟左右。。

最终的结果文件大概生成如下图,

image.png

在这里frozen_graph文件夹下面有用于在310推理侧的文件,我们来试试看训练出来的效果如何~

四、推理

首先需要将生成的结果做一个pbom模型的转换,

image.png

转换的输入目录选择frozen_graph,输出选择目录om,这些均可参考算法描述的文档,最终转换完成后在ommodel目录下会出现一个om模型。

之后再导入模型,选择从模板中导入,目录选择om/model下的目录,image.png

image.png

创建好后直接部署在线服务即可,预测一下看看效果,还是不错的哈哈


image.png

【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。