YOLO2安装教程

本博客详细介绍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

测试效果
说明大功告成。

原文地址：https://www.jb51.cc/ubuntu/350438.html