win7配置darknet使用darknet进行yolo3训练自己的数据集

win7配置darknet使用darknet进行yolo3训练

  • 1、说明:
  • 2、配置darknet
  • 3、使用YOLO3进行训练自己的数据集
    • 结束

1、说明:

我主要是参考其他博主的文章,所以我贴出链接,主要说以一些关键步骤即可;

2、配置darknet

我用的是和这篇博主一样的环境,TensorFlow-gpu1.5+cuda9.0+cudaa7.1+vs2015。
配置需要注意的是:先安装vs2015,然后安装cuda和cudnn,如果要用到opencv需要在系统环境path配置opencv;然后使用relwase 和X64生成即可。(我在生成后就提示opencv找不到,主要就是环境配置路径不对,重新配置就ok;还有各bug就是报错include下缺少文件,按需要把cuda下的include文件下的文件复制到vs对应的include文件下(这个应该就是我先安装的cuda再安装vs2015埋下的坑吧)。因为我是配置好了决定记录下这个问题,所以只有思路,没有图片,抱歉!)
参考链接:https://blog.csdn.net/baidu_36669549/article/details/84501759

3、使用YOLO3进行训练自己的数据集

我之所以使用darknet进行训练,主要是遇到一个bug,不知道怎么处理,我使用开始是在我的一台cpu笔记本上搭建的yolo3环境,训练自己的数据集,一切正常(但是训练速度太慢了,一天都没跑完,最后放弃了),因为训练太耗时间了,所以我放弃了,准备在GPU电脑训练,我把环境搭建好之后,开始训练,问题来了o(╥﹏╥)o程序正常运行,输出Epoch 1/10信息后,程序就没有下文了,不报错,不进行训练,我的内存直线增长,搞了一周没找到原因,放弃了(各位大佬有知道的可以指点小弟一下,感谢!)win7配置darknet使用darknet进行yolo3训练自己的数据集_第1张图片
**下面开始说darknet训练:**本人小白,所以离不了巨人的肩膀,还是贴出大佬的链接yolo3训练自己的数据集https://www.cnblogs.com/answerThe/p/11481564.html,我是在window上进行的,参考这篇博客就ok了,需要注意的是最后执行的指令需要改以下,就是进入到自己生成的darknet.exe文件夹下,执行指令darknet.exe detector train cfg/my_data.data cfg/my_yolov3.cfg darknet53.conv.74,我直接执行原指令./darknet detector train cfg/my_data.data cfg/my_yolov3.cfg darknet53.conv.74会报错没有“.”文件,还要注意下,对应自己的data路径也要改以下,基本注意的就是这些。最后贴一张我自己训练出来的结果吧,主要是检测两个零件的放置状态,结果还是OK的。
win7配置darknet使用darknet进行yolo3训练自己的数据集_第2张图片
(可以看出这个框选的位置还不是太好,所以接下来去把yolo3再优化下。)

结束

你可能感兴趣的:(笔记)