版權聲明:本文爲博主原創文章,歡迎轉載,並請註明出處。聯繫方式: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
運行結果顯示