本博客详细介绍YOLO2安装的步骤(默认ubuntu已经安装),跟部分版本如下:
Ubuntu:16.04 destop
Opencv:3.3.0
CUDA:8.0
CUDNN:6.0
ubuntu安装参考博客:http://www.jb51.cc/article/p-pntjumus-bpw.html
ubuntu基本应用参考博客:http://www.jb51.cc/article/p-zeygkjtz-bpw.html
安装参考博客(cuda):http://www.jb51.cc/article/p-vecitgns-bpw.html
正式安装
配置darknet
git clone https://github.com/pjreddie/darknet
cd darknet
make
看到如下输出就是表示成功
mkdir -p obj
gcc -I/usr/local/cuda/include/ -Wall -Wfatal-errors -Ofast....
......
gcc -I/usr/local/cuda/include/ -Wall -Wfatal-errors -Ofast -Im.....
如果没有报错输入
./darknet
得到输出
usage: ./darknet <function>
说明darknet配置成功
打开darknet目录下Makefile文件,修改代码
GPU=1<span style="white-space:pre"></span>
CUDNN=1
OPENCV=1
保存完成后重新编译。就可以实现基于cuda和opencv的编辑
./darknet
下载预训练模型
wget https://pjreddie.com/media/files/yolo.weights
也可以从该链接直接下载:https://pjreddie.com/darknet/yolo/
测试
在darknet目录下新建一个weights,将下载好的训练文件放到该目录下。(也可以将目录记清楚不用新建)
cd darknet
./darknet detect cfg/yolo.cfg weights/yolo.weights data/dog.jpg
测试效果
说明大功告成。