windows版的darknet安装和使用(深度学习中目标检测框架)

1、系统环境

  1. windows: win7 x64 ultimate
  2. cpu: i5 5257u
  3. gpu: 940M
  4. cuda: 9.0
  5. note: 系统环境对darknet的安装和使用影响不大,只要是windows系统都可以适用,需要注意cuda版本

2、darknet安装

2.1、下载地址(github上的源码)

github上的源码,需要自行编译。链接: https://github.com/pjreddie/darknet
已经编译过的darknet源码下载地址。链接: https://pan.baidu.com/s/15gsbnEKzJVgyfRX9nlzOxw 提取码: 2emz

2.2、安装步骤

  1. 工具安装
    1.1 安装VS2015 update3,用来进行编译
    1.2 安装cuda9.0(可选)
    1.3 安装cudnn(可选,若安装了cuda,则建议安装相应版本的cudnn)
    1.4 ps: 以上软件的安装教程见该链接前半部分https://blog.csdn.net/u011609063/article/details/84188942
  2. darknet编译
    2.1 将下载的darknet解压缩,我的是在d盘根目录下,若是下载了已编译的版本则无需进行以下步骤
    在这里插入图片描述
    2.2 进入D:\darknet-master\build\darknet目录下,双击darknet.vcxproj,使用nodepad++打开
    在这里插入图片描述
    搜索cuda,将9.0改为自己已经安装的cuda版本,共两处。GitHub源码上是10.0,我的是9.0,这个请对应你们自己的cuda版本。修改位置大概在55行和291行。
    在这里插入图片描述
    在这里插入图片描述
    2.3 进入D:\darknet-master\build\darknet目录下,双击darknet.sln
    在这里插入图片描述
    注意修改为Release, x64,如图。右击darknet,点击build或生成。
    在这里插入图片描述
    2.4 若在D:\darknet-master\build\darknet\x64下生成了darknet.exe,则编译成功,否则请对vs命令行中错进行修改或者评论提问。

3、darknet使用

打开命令行,定位到darknet.exe所在目录,我的是D:\darknet-master\build\darknet\x64,运行以下命令。
说明:

  1. test表示对图片进行检测和识别,demo表示对视频进行检测是和识别。
  2. -i int_number表示使用第(int_number + 1)个gpu进行计算,例如-i 0表示使用第一个gpu进行计算
  3. -thresh thresh_value表示当thresh>=thresh_value时检测成功。thresh_value∈[0, 1]
  4. []表示可选。
  5. -out_filename output_path_and_name表示将结果保存

3.1、图片的检测与识别

通用模板:
darknet.exe detector test data_file_path_and_file cfg_file_path_and_name weights_file_path_and_name
[-i int_number] [-thresh thresh_value] photo_path_and_name [-ext_output]

example: 
darknet.exe detector test data/coco.data cfg/yolov3.cfg yolov3.weights -i 0 -thresh 0.25 data/dog.jpg -ext_output
或
darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg

执行了之后会在darknet.exe的同目录下生成preditions.jpg文件,这个表示检测结果。

3.2、视频的检测与识别

视频检测:

通用模板:
darknet.exe detector demo data_file_path_and_name cfg_file_path_and_name weights_file_path_and_name 
video_name_path_and_file [-i int_number] [-out_filename output_path_and_name]

不保存视频:

example: 
darknet.exe detector demo data/coco.data cfg/yolov3.cfg yolov3.weights data/test.mp4 -i 0
这个不会生成结果,若安装了opencv则可以直接看到结果,否则的话会没有反应

保存视频:

example: 
darknet.exe detector demo data/coco.data cfg/yolov3.cfg yolov3.weights data/test.mp4 -i 0 -out_filename res.MP4

写博客不易,转载请注明出处。

  • 9
    点赞
  • 81
    收藏
    觉得还不错? 一键收藏
  • 60
    评论
Windows安装YOLOv3 (You Only Look Once) 和 Darknet需要几个步骤。YOLOv3是一个基于Darknet框架目标检测算法,而PyCharm是一款流行的Python集成开发环境。以下是安装过程: 1. **安装Python**:首先确保已安装最新本的Python(建议使用Python 3.6及以上),你可以从Python官网下载并安装。 2. **安装Darknet依赖**:由于Darknet是用C++编写的,你需要C编译器如Microsoft Visual Studio(MSVC)或MinGW。访问https://github.com/AlexeyAB/darknet 下载源码,解压后按照说明配置编译环境。 3. **编译Darknet**:在命令行进入Darknet目录,运行`./configure`,然后选择合适的编译选项(例如`make -j8` 或 `mingw32-make`),最后执行`make`来编译Darknet库和工具。 4. **安装PyTorch或TensorFlow**:YOLOv3可以与多种深度学习库结合,如PyTorch。如果你计划使用PyTorch,安装`torch`, `torchvision`, 和 `torch.utils.ffi`。如果你更倾向于TensorFlow,安装`tensorflow`和`opencv-python`。 5. **安装YOLOv3 for Python**:有两种方式,一是将Darknet的C语言API整合到Python项目,这通常涉及到编译时链接;二是直接使用预训练模型,比如通过`yolov3.pytorch`这样的GitHub项目。你可以选择下载预训练权重并将其导入到PyCharm。 6. **在PyCharm设置项目**:创建一个新的Python项目,在项目结构添加必要的文件夹,如包含模型的`models`目录,并配置Python解释器路径。 7. **编写Python脚本**:编写Python脚本来加载模型、处理图像以及进行预测。PyCharm会提供强大的调试和代码提示功能。
评论 60
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值