yolo 学习系列（一）：安装与测试

yolo 学习系列（一）：安装与测试

重点参考
官方教程
初学JetsonTX2之部署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 https://pjreddie.com/media/files/yolov3.weights  #普通版yolo
wget http://pjreddie.com/media/files/tiny-yolo-voc.weights #tiny版yolo

2.2 权值测试

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

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

# 测试视频
./darknet detector demo cfg/coco.data cfg/yolov3.cfg yolov3.weights data/test.mp4

为了便于命令的输入，可将以上每条命令写入一个 .sh 的脚本文件中，然后运行脚本文件即可。