[blog8]ubuntu18.04配置yolov3

参考https://blog.csdn.net/weixin_42747301/article/details/96636614#_76

前提:如果想用GPU,装好N卡驱动,cuda cudnn opencv再来

下载darknet,下载yolove.weights

git clone https://github.com/pjreddie/darknet
cd darknet
make
wget https://pjreddie.com/media/files/yolov3.weights

在cfg里面把yolov3.vfg改为test的batch和sub=1,train注释掉,测试

如果测试报错out of memory,就把cfg里面的图像大小改一改

./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg

出现结果:

layer     filters    size              input                output
    0 conv     32  3 x 3 / 1   416 x 416 x   3   ->   416 x 416 x  32  0.299 BFLOPs
    1 conv  

……

Loading weights from yolov3.weights...Done!
data/dog.jpg: Predicted in 0.142452 seconds.
dog: 99%
truck: 83%
car: 51%
bicycle: 100%

这个时候是没有GPU啥的

修改darknet目录下的makefile,

GPU=1
CUDNN=1
OPENCV=1
OPENMP=0
DEBUG=0

ARCH根据自己电脑性能改,实在不知道就用默认的

NVCC=/usr/local/cuda-9.0/bin/nvcc

COMMON+= -DGPU -I/usr/local/cuda-9.0/include/
CFLAGS+= -DGPU
LDFLAGS+= -L/usr/local/cuda-9.0/lib64 -lcuda -lcudart -lcublas -lcurand

改好了之后

sudo make clean

sudo make -j8

重新用狗狗图片测试(就刚才那个命令),出结果就说明配置好了

猜你喜欢

转载自blog.csdn.net/Drinks_/article/details/103855291