版权声明:本文为博主原创文章,欢迎转载,并请注明出处。联系方式:460356155@qq.com
一、准备工作:
安装pkg-config:sudo apt-get install pkg-config
安装opencv:sudo apt-get install libopencv-dev
二、编译darknet:
下载 darknet-master.zip:https://github.com/pjreddie/darknet,解压
修改Makefile:
GPU=1
CUDNN=1
OPENCV=1
OPENMP=1
DEBUG=1
把cudnn的安装路径添加到COMMON和LDFLAGS中:
ifeq ($(GPU), 1)
COMMON+= -DGPU -I/usr/local/cuda/include/ -I/home/zbq/cuda/include/
CFLAGS+= -DGPU
LDFLAGS+= -L/usr/local/cuda/lib64 -L/home/zbq/cuda/lib64 -lcuda -lcudart -lcublas -lcurand
endif
编译:make
三、预训练模型测试
下载yolov3.weights
:https://pjreddie.com/media/files/yolov3.weights
./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
运行结果显示