您的位置:首页 > 其它

YOLO2安装教程

2017-11-28 12:27 218 查看
本博客详细介绍YOLO2安装的步骤(默认ubuntu已经安装),跟部分版本如下:

Ubuntu:16.04 destop

Opencv:3.3.0

CUDA:8.0

CUDNN:6.0

ubuntu安装参考博客:http://blog.csdn.net/ezhchai/article/details/52891014

ubuntu基本应用参考博客:http://blog.csdn.net/ezhchai/article/details/52972064

安装参考博客(cuda):http://blog.csdn.net/ezhchai/article/details/69663082

正式安装

配置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[/code] 
也可以从该链接直接下载:https://pjreddie.com/darknet/yolo/



测试

在darknet目录下新建一个weights,将下载好的训练文件放到该目录下。(也可以将目录记清楚不用新建)

cd darknet
./darknet detect cfg/yolo.cfg weights/yolo.weights data/dog.jpg


测试效果



说明大功告成。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  cuda ubuntu