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