Linux环境下Darknet的安装与测试

1.下载Darknet源码

sudo git clone https://github.com/pjreddie/darknet

2.编译Darknet源码

cd darknet

make

3.下载权重文件yolov3.weights,放入到darknet文件夹下

下载地址:

https://pjreddie.com/media/files/yolov3.weights

4.修改cfg文件夹下的yolov3.cfg文件,原文件是训练模式,需要改成测试模式。

具体修改方法是,打开Testing下的两个参数,注销掉Training下的两个参数

Linux环境下Darknet的安装与测试_第1张图片

5.运行网络,查看检测效果

./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg

可以看出网络运行时间非常慢,要12s多。这主要是因为目前网络是在CPU上运行,没有用到GPU。

Linux环境下Darknet的安装与测试_第2张图片

6.修改Makefile文件,打开GPU的开关,将GPU=0改为GPU=1,并重新编译。

Linux环境下Darknet的安装与测试_第3张图片

7.再次执行

./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg

这下速度快多了

Linux环境下Darknet的安装与测试_第4张图片

同时可以看出,GPU跑起来了。

Linux环境下Darknet的安装与测试_第5张图片

你可能感兴趣的:(计算机视觉,编程相关)