yolo 的安装和使用

yolo的安装和使用

重点参考
官方教程

1 yolo下载与安装

1.1 yolo下载

在shell终端输入以下命令：

# 克隆下载速度很慢，可进入该网址下载安装包，解压使用
git clone https://github.com/pjreddie/darknet.git
cd darknet
gedit Makefile     #修改makefile

1.2 修改配置文件

博主采用的是笔记本，宏定义修改如下（其余不用修改）：

GPU=0           # 不使用 GPU 加速
CUDNN=0         # 若支持cudnn加速，则为1
OPENCV=1        # 使用opencv
DEBUG=0

保存退出

1.3 编译验证

yolo使用过程中的命令行都是在 ～/darknet $ （文件一级目录）下执行的
Caffe 也遵循以上规则

在 darknet/ 一级目录下编译darknet，输入以下命令：

make

编译完成后生成可执行文件 darknet（菱形紫色标志），验证

./darknet
#若输出以下语句，则为安装完成
usage: ./darknet <function> 

2 yolo的使用

2.1 权值下载

下载官方已经训练好的模型

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

2.2 权值测试

# 测试图片
./darknet detector test cfg/coco.data cfg/yolo.cfg yolo.weights test.jpg

# 测试摄像头
./darknet detector demo cfg/coco.data cfg/yolo.cfg yolo.weights

# 测试视频
./darknet detector demo cfg/coco.data cfg/yolo.cfg yolo.weights <video file>