一、基于海思芯片:从算法到移植 darknet框架下训练yolov3

今天是2020年1月25号,大年初一,万万没想到,这个本来狂舞躁动的时刻,敝人会安逸的坐在家里写博客,这场肺炎带给我们的应该是深刻反思。边缘计算如日中天,或者说把深度学习落地,搞到移动端解决是一个大趋势,现实情况是多数搞算法的对嵌入式方面的移植不甚了解,而搞嵌入式的对这个高大上的算法环节更是一脸懵逼。下面我就带领大家梳理一下,认真研读完本系列四篇教程并下载所有资料,帮助你从训练算法一直到移植海思3516dv300,畅通无阻,实现快速开发迭代。好吧,进入正题,本系列会以四个篇章讲解完毕yolov3从训练到移植到海思芯片,如下所示:

一、基于海思芯片:从算法到移植   darknet框架下训练yolov3

二、基于海思芯片:从算法到移植   darknet框架下yolov3转为caffe框架下

三、基于海思芯片:从算法到移植   caffe-yolov3测试训练效果

四、基于海思芯片:从算法到移植   生成海思wk文件并进行成果展示

 

针对一个全新的所识别目标,例如电瓶车、安全帽、反光服、手套、火苗、吸烟检测、烟雾检测等等,诸如此类,首先采集和标注图像,在windows上可参照网址https://cloud.tencent.com/developer/news/325876,标注的图像格式有.xml格式和txt格式,根据需要可以抓换,稍后会把.xml格式转为.txt格式代码一并奉上,也可网上自行参考。顺便提一下,图像可以自己采集也可以网上下载,现在图片估计是不好爬了,我两三年没写过爬虫了,不了解现在的行情,我还是给大家提一下,你最好在标注前把图片统一按规则命名一下,不为别的,显得咱爷们姐妹专业一点,代码我有,自己写也不难,稍后也会上传。

针对darknet训练的前期环境配置我就不再赘述,不然篇章过于冗余,地址为https://github.com/pjreddie/darknet,自行下载安装配置,关于darknet的配置博客不少,只需要记住把opencv和cuda安装正确方可使用,后面我会把所有的资源源代码上传,大大缩减大家的开发时间。我自己的darknet文件以及训练的一些目标物集合如图所示

而对于为什么选择yolov3这个网络,实在是因为很经典,在工业上应用较多,当然你也可以在caffe下搭载自己的网络,但是我强烈不建议,因为你现在要做的是把算法到移植海思整个流程跑通,以后有时间会和大家深入交流深度学习,咱当年可是从BP神经网络一点点推导的过来人。yolov3的官网说明:YOLO: Real-Time Object Detection  官网地址https://pjreddie.com/darknet/yolo/,说的甚是详细,当然网络上也有很多详细讲解yolov3的博客,都可以拿来为我所用。我在网上给大家选择了一个通俗易懂的篇章,如何在darknet框架下训练yolov3 可以参考网址链接https://www.cnblogs.com/answerThe/p/11481564.html,讲的还不错 ,帮助你训练起来自己的数据集。训练自己的数据集后,其实只要迭代几千次,你就可以大概测试一下自己的网络效果,你自己设置路径会在backup中生成大量的.weights文件,还有配合你自己的.cfg文件,这两个文件会构成你接下来所需要的所有信息,举个例子,我自己训练安全帽,通过训练得到anquanmao_yolov3-voc_30000.weights权值文件,还有yolov3.cfg文件,yolov3.cfg文件要根据你自己的显卡能力,还有目标物的多少自己设置相关的参数,在我文中的链接中都会提到如何设置,不再赘述。anquanmao_yolov3-voc_30000.weights和yolov3.cfg将为下一篇中转换生成caffe下的caffemodel和prototxt备用,另外再提一句,无论在什么框架下,所生成的权值文件都需要测试一下效果,再进行进一步的操作,在darknet文件中可以进行下面的操作,在编译后在darknet文件夹中,运行命令

./darknet detect cfg/yolov3.cfg backup/anquanmao_yolov3-voc_30000.weights data/0.jpg

要根据你自己的情况更改cfg和weights文件,以及要测试的图片,如果是项进行实时测试,查个小摄像头,执行命令

./darknet detector demo cfg/anquanmao.data backup/anquanmao_yolov3-voc_30000.weights

在本篇中已经生成anquanmao_yolov3-voc_30000.weights,以及改写的yolov3.cfg,将为下一篇中转换生成caffe下的caffemodel和prototxt备用。另外在本篇结束后,包括安全帽在内的所有的框架下的训练权值文件都会让小伙伴下载,也请大家支持我的团队。

 

发布了7 篇原创文章 · 获赞 2 · 访问量 3183

猜你喜欢

转载自blog.csdn.net/Bonjour_ca_va/article/details/104084984