说明: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 yolo.weights -c 1 //测试USB相机
./darknet detector demo cfg/coco.data cfg/yolov2.cfg yolo.weights rtsp://admin:12345@172.6.22.106:554/h264/ch1/main/av_stream //网络相机,使用网络相机时TX2网段与相机要一致