YOLOv3初体验

网友投稿 588 2022-05-29

一、算法订阅:

首先登录华为云ModelArts的AI市场,

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

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

二、准备数据集

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

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

OBS Browser-:https://support.huaweicloud.com/browsertg-obs/obs_03_1003.html

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

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

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

三、训练

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

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

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

经过大概10分钟左右。。

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

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

四、推理

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

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

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

YOLOv3初体验

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

人工智能

版权声明:本文内容由网络用户投稿,版权归原作者所有,本站不拥有其著作权,亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容,请联系我们jiasou666@gmail.com 处理,核实后本网站将在24小时内删除侵权内容。

上一篇:【2020华为云AI实战营】 第3章:图像分割+3.2小节自己动手的过程
下一篇:【超级账本】Fabric 层次结构以及核心模块的介绍(二)
相关文章