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下的两个参数

5.运行网络,查看检测效果
./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
可以看出网络运行时间非常慢,要12s多。这主要是因为目前网络是在CPU上运行,没有用到GPU。
6.修改Makefile文件,打开GPU的开关,将GPU=0改为GPU=1,并重新编译。
7.再次执行
./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
这下速度快多了
同时可以看出,GPU跑起来了。