说明:cuda、cudnn、opencv等环境均已经配置好(详见前面的博客)
从github上下载darknet:https://github.com/pjreddie/darknet
之后编译darknet
1、首先修改makefile文件,打开GPU、CUDNN和OPENCV。
GPU=1
CUDNN=1
OPENCV=1
OPENMP=0
DEBUG=0
2、执行make
,编译过程如下图
3、测试yolo
./darknet detect cfg/yolov2.cfg yolo.weights data/person.jpg //测试图片
./darknet detector demo cfg/coco.data cfg/yolov2.cfg -c 1 //测试USB相机
./darknet detector demo cfg/coco.data cfg/yolov2.cfg rtsp://admin:[email protected]:554/h264/ch33/main/av_stream //网络相机