darkNet yolo+vs2017+cuda10.1+win10配置(详细)

简介:

  • darknet是用C语言写的一个开源的深度学习框架,支持CPU、GPU。它不像tensorflow那么庞大,对于使用C/C++开发的工程也很容易嵌入进去。
  • 本篇文章配置的环境是win10,vs2017,cuda10.1,opencv3.4.1,其中vs其他版本vs2015、vs2019的配置也是一样的,opencv只要高于2.4就可以,cuda其他版本的话本文中的2.5条那里的compute_75,sm_75需要改一下,现在我不知道cuda版本和它的对应关系,原项目是cuda11.1的,对应compute_86,sm_86.(如果有人知道了还请赐教)
  • 用darknet yolov4训练自己的数据集的参考链接
  • github:Yolo v4, v3 and v2 for Windows and Linux

文章目录

      • 1. [下载](https://github.com/AlexeyAB/darknet)
      • 2. 配置
        • 2.1 修改windowsSDk和平台工具集
        • 2.2 配置opencv和cuda附加包含目录
        • 2.3 修改cuda版本
        • 2.4 CUDA路径配置
        • 2.5 编译出现MSB372错误
        • 2.6 编译出现无法打开“opencv_world341.lib”
      • 3. 编译
      • 4. 使用
        • 4.1 下载预训练模型
        • 4.2 命令行下运行
  • C o n g r a t u l a t i o n s ! 至 此 , 所 有 配 置 完 成 ! \color{#f25612}{Congratulations!至此,所有配置完成!} Congratulations!
        • 4.3 更多命令行用法
      • 附其他检测结果

1. 下载

darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第1张图片
用git下载或者手动下载都一样,下载下来是这样
darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第2张图片

2. 配置

  • 打开build,用vs2017(或者其他版本的vs,原来的项目是用vs2015的)打开darknet.sln
    darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第3张图片

2.1 修改windowsSDk和平台工具集

  • 原设置
    darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第4张图片
  • 修改后
    darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第5张图片

2.2 配置opencv和cuda附加包含目录

  • 原属性页的C/C++ 附加包含目录
    darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第6张图片
  • 修改后
    darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第7张图片

2.3 修改cuda版本

注:在属性页上面没有看到cuda的配置选项,则需要手动添加:属性–生成依赖项–生成自定义。修改后可以与2.4中的“cuda-common.png“图上的属性页对比,可以看到属性页上多了CUDA C/C++和CUDA Linker选项
darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第8张图片

  • 原设置
    darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第9张图片
  • 修改后
    darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第10张图片

2.4 CUDA路径配置

darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第11张图片

2.5 编译出现MSB372错误

参考链接中的第4点
darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第12张图片

  • 解决方法:打开darknet.vcxproj(就在darknet.sln同级目录下)修改

      64
      compute_35,sm_35;compute_86,sm_86
    

改为


      64
      compute_35,sm_35;compute_75,sm_75
    

2.6 编译出现无法打开“opencv_world341.lib”

darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第13张图片

  • 这是opencv库目录配置的问题,解决方法:属性—vc++目录—库目录,添加opencv的lib文件夹路径。
    darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第14张图片
    连接器—输入—附加依赖项 添加“opencv_world341.lib”
    darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第15张图片

3. 编译

注意配置的模式和生成模式要对应,如:属性页配置的是release x64的,生成时也要是release x64模式。darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第16张图片

4. 使用

4.1 下载预训练模型

地址
以yolov4.weight为例
darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第17张图片
yolov4.weight下载后放到编译生成的darknet.exe同级目录下,一般是build\darknet\x64下
darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第18张图片

4.2 命令行下运行

win+r打开cmd命令行模式,用cd进入到darknet.exe的目录
输入以下命令

darknet.exe detector test cfg/coco.data cfg/yolov4.cfg yolov4.weights -thresh 0.25

darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第19张图片
darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第20张图片

输入图片路径data/person.jpg
darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第21张图片

C o n g r a t u l a t i o n s ! 至 此 , 所 有 配 置 完 成 ! \color{#f25612}{Congratulations!至此,所有配置完成!} Congratulations!

4.3 更多命令行用法

链接

  • 附其他检测结果

不为别的只是想体验一下

darkNet yolo+vs2017+cuda10.1+win10配置(详细)_第22张图片


你可能感兴趣的:(machine,learning,深度学习,计算机视觉)