目标检测之三 Win10 +VS2015+双gtx Titan X使用YOLO v2训练VOC数据

本文详细介绍了在Win10系统、VS2015和双GPU环境下,如何使用YOLO v2进行VOC数据集的训练。内容包括环境配置、CUDA和CUDNN的安装、darknet源码的编译以及训练过程,强调了训练过程中需要注意的参数调整和文件配置。
摘要由CSDN通过智能技术生成

Win10 +VS2015+双gtx Titan X使用YOLO v2训练VOC数据

其他博客地址:http://blog.csdn.net/baolinq

    本文主要是记录在win10下,yolo v2训练数据的一整个过程,网上很多是在linux下的,不得不说这个东西还是linux下方便啊,主要依赖项:

VS2015或者VS2013都可以

Opencv2.x或者opencv3.x

Cuda8.0和cudnn6.0

GitHub上下载darknet源码

最近老师给配了一个深度学习专用的电脑,两路Titan X显卡+i7 6700K+SSD 256G+HD 4T(企业级),自己组装,搞了一下午(电脑硬件小白)。然后装双系统,配环境,真的是麻烦,搞了很久。磨刀不误砍柴工,为了以后效率高点还是值得的。

    好了,不扯淡了。回到正题,使用YOLO v2训练VOC数据,我觉得使用yolo的朋友应该都是看上了它的速度,可以做到实时检测,这是最大的优点。

1.  配环境

装cuda和cudnn环境,这里就不说了,直接官网下载下来一路默认安装就好了,本文使用cuda8.0和cudnn6.0,cudnn下载的时候需要注册和填一个调查问卷,很简单这里就不谈了,网上也有很多是写这个的。配好在cmd窗口运行nvcc –V,有出现


表示安装成功。

为了显示图像,这里还安装了opencv,随便选一个版本,我装了opencv2.413,具体安装方法见我转载的博客(),并把环境配好,主要是环境变量和VS是属性管理器。

 

2.  编译源文件

从GitHub上下载darknet网络的源代码,在darknet-master\build\darknet目录下有这些文件,包括gpu版本和cpu版本,我们这里选择darknet.vcxproj,即是gpu版本,用VS

打开一般是出错的,不能正确加载项目,这是正常的。如果是no gpu版本的是能正常打开的,它并不依赖cuda的环境。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值