YOLO V3-GPU版本在Windows配置及注意事项

0.首先还是确保VS2015 成功安装 配置好opencv320,这是我用得版本

官方写得(很清楚)也是VS2015,opencv的版本有要求,要≤3.4.0。还有就是CUDA我用的是CUDA9.0,官方最新的源码用的是CUDA10.0。

配置好cudnn。这些内容很多,不赘述。

1.修改darknet.vcxproj文件,将CUDA版本修改成自己的,我的是9.0


  
    
  
   ...... 
   ...... 
   ...... 
   ...... 
   ...... 
   ...... 
 
    
  

2.打开darknet.sln

x修改为---Release ---x64

YOLO V3-GPU版本在Windows配置及注意事项_第1张图片

属性---VC++目录

                    ---包含目录

                    --- 库目录

YOLO V3-GPU版本在Windows配置及注意事项_第2张图片

 链接器---输入附加依赖项opencv_world320.lib

3.点击生成---生成解决方案

YOLO V3-GPU版本在Windows配置及注意事项_第3张图片

成功生成

YOLO V3-GPU版本在Windows配置及注意事项_第4张图片

4. 训练(没有自己的数据集可以跳过直接到5)

详细的可以参考这篇YOLO-V3训练中会遇到的问题

 

输入

darknet.exe detector train data/voc.data cfg/yolov3-voc.cfg

也可以用预训练权重

darknet.exe detector train data/voc.data cfg/yolov3-voc.cfg darknet53.conv.74

训练过程

YOLO V3-GPU版本在Windows配置及注意事项_第5张图片

5.测试

输入

darknet.exe detector test data/coco.data cfg/yolov3.cfg yolov3.weights dog.jpg

 结果:

YOLO V3-GPU版本在Windows配置及注意事项_第6张图片

6. 修改源码(添加自己想要的额外功能——比如YOLO-V3将检测的目标单独保存)

可以参考链接中的博客。

以上内容是受委托添加,有误请指正。

邮箱1:[email protected]

邮箱2:[email protected]

QQ不经常在线,见谅~

你可能感兴趣的:(YOLOv3,yolo,目标检测,YOLO目标检测)