YOLOv4在windows下的安装配置

YOLOv4

安装依赖

括号里是我当前版本

  • vs(vs2015)
  • cuda(cuda10.1)
  • cudnn(cudnn7)
  • opencv(opencv4.1)
  • cmake(cmake3.15)

安装步骤

1.生成项目

  • 打开cmake,设置源文件路径(项目根目录)与输出文件路径(自定义例如mybulid)

  • 通过添加与删除配置选项如下图所示
    YOLOv4在windows下的安装配置* 点击Configure,弹出对话框如下,选择编译器(Visual studio 14 2015)与平台(x64)
    YOLOv4在windows下的安装配置

  • 点击Finish后开始运行,显示Configuring Done后点击Generate生成项目

2.编译可执行文件

  • 进入输出文件目录,双击Darknet.sln运行vs
    YOLOv4在windows下的安装配置

  • 选择Release x64配置运行编译
    YOLOv4在windows下的安装配置

  • 运气不好可能会遇到无法访问,需要将右边的各个解决方案一个个编译

3.运行可执行文件

  • 进入Release目录下,将项目跟目录下的data文件夹cfg文件夹3rdparty\pthreads\bin\pthreadVC2.dll和下载下来的权重文件放到Release目录
    YOLOv4在windows下的安装配置

  • 运行darknet.exe detector demo cfg/coco.data cfg/yolov4.cfg yolov4.weights -c 0使用摄像头进行检测,可保存为bat脚本一键执行

总结

  • yolov4显存要求较高,笔记本的2G显卡会显存溢出
  • 可修改bat文件使用不同的模型进行测试,例如darknet.exe detector demo cfg/coco.data cfg/yolov3-tiny-prn.cfg yolov3-tiny-prn.weights -c 0使用yolov3-tiny模型进行检测

YOLOv4在windows下的安装配置