yolo的安装和使用

一.前提

安装opencv：用来输出图像
安装cuda：使用GPU加速
也可参考官方安装步骤：http://pjreddie.com/darknet/install/

二.下载安装darknet

shell终端

git clone https://github.com/pjreddie/darknet.git
cd darknet
gedit Makefile #修改makefile

修改成下面的代码

GPU=1 #安装好cuda，并打算使用GPU。
CUDNN=0 #若支持cudnn加速，则为1
OPENCV=1#使用opencv
DEBUG=0

保存退出，编译darknet，shell终端

make

若编译出现类似如下内容：

gcc -I/usr/local/cuda/include/  -Wall -Wfatal-errors  -Ofast....
gcc -I/usr/local/cuda/include/  -Wall -Wfatal-errors  -Ofast....
gcc -I/usr/local/cuda/include/  -Wall -Wfatal-errors  -Ofast....
.....
gcc -I/usr/local/cuda/include/  -Wall -Wfatal-errors  -Ofast -lm....

不用担心，shell终端

./darknet

#若输出：usage: ./darknet <function> 则为安装完成

三.编译报错解决办法

1.报错类似：/usr/bin/ld 找不到 -lippicv
此类错误属于opencv安装问题，见
opencv安装步骤

2.报错：cuda runtime 问题
cuda安装错误，缺少toolkit。请卸载cuda，使用runfile安装cuda
安装cuda的错误解决办法

3.报错：无法找到libcudart.so.7.5
解决办法：在此目录下
shell终端

export LD_LIBRARY_PATH=/usr/local/cuda7.5/lib64:$LD_LIBRARY_PATH

四.使用yolo（v2）

1.官方使用介绍http://pjreddie.com/darknet/yolo/
2.下载权值
shell终端

wget http://pjreddie.com/media/files/yolo.weights #普通版yolo
wget http://pjreddie.com/media/files/tiny-yolo-voc.weights #tiny版yolo

3.测试图片
shell终端

./darknet detector test cfg/coco.data cfg/yolo.cfg yolo.weights test.jpg

4.测试摄像头
shell终端

./darknet detector demo cfg/coco.data cfg/yolo.cfg yolo.weights

5.测试视频
shell终端

./darknet detector demo cfg/coco.data cfg/yolo.cfg yolo.weights <video file>