ubuntu 18 安装YOLO V3并测试摄像头

1. 安装darknet

$ git clone https://github.com/pjreddie/darknet
$ cd darknet

添加GPU及OpenCV支持:

$ vi Makefile

修改如下:

GPU=1
CUDNN=1
OPENCV=１ # 如果安装了opencv则改为１
OPENMP=0
DEBUG=0
...
NVCC=/usr/local/cuda-9.0/bin/nvcc # 原为 NVCC = nvcc，cuda后面的紧跟自己的cuda版本，如cuda-9.0

$ make -j4

2. 下载预训练权重

$ wget https://pjreddie.com/media/files/yolov3.weights

3. 测试

测试图片:

$ ./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg

测试摄像头:

$ ./darknet detector demo cfg/coco.data cfg/yolov3.cfg yolov3.weights

如果出现Error: out of memory:
修改cfg文件夹下的yolov3.cfg，把Testing的batch和subdivisions改为1，width和height改小：

[net]
# Testing
batch=1
subdivisions=1
# Training
# batch=64
# subdivisions=16
width=416
height=416