版权声明:本文为博主原创文章,欢迎转载,并请注明出处。联系方式:460356155@qq.comgit
1、准备工做:github
安装pkg-config:sudo apt-get install pkg-config测试
安装opencv:sudo apt-get install libopencv-devspa
2、编译darknet:code
下载 darknet-master.zip:https://github.com/pjreddie/darknet,解压blog
修改Makefile:ip
GPU=1
CUDNN=1
OPENCV=1
OPENMP=1
DEBUG=1
把cudnn的安装路径添加到COMMON和LDFLAGS中:get
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
endifit
编译:make编译
3、预训练模型测试
下载yolov3.weights
:https://pjreddie.com/media/files/yolov3.weights
./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
运行结果显示