预备工作:ubuntu下完成了cuda/cudnn/opencv3.2的编译工作
1. yolov3下载
地址:https://github.com/pjreddie/darknet.git
YOLO官网:https://pjreddie.com/darknet/yolo/
2.yolov3安装
修改Makefile文件
GPU=1 #如果使用GPU设置为1,CPU设置为0
CUDNN=1 #如果使用CUDNN设置为1,否则为0
OPENCV=1 #如果调用摄像头,还需要设置OPENCV为1,否则为0
OPENMP=0 #如果使用OPENMP设置为1,否则为0
DEBUG=0 #如果使用DEBUG设置为1,否则为0
ARCH= -gencode arch=compute_30,code=sm_30 \
-gencode arch=compute_35,code=sm_35 \
-gencode arch=compute_50,code=[sm_50,compute_50] \
-gencode arch=compute_52,code=[sm_52,compute_52] \
-gencode arch=compute_60,code=[sm_60,compute_60] \
-gencode arch=compute_61,code=[sm_61,compute_61] #Titan是pascal架构
…
CC=gcc
NVCC=/home/user/cuda-9.0/bin/nvcc #修改为自己的路径
注意问题:在添加gpu性能参数时,复制时别忘了加**\**
不然make时会报错:
bobo@bobo-System-Product-Name:~/darknet$ make
Makefile:11: *** missing separator。 停止。
若出现错误,键入
make clean
纠正错误后,然后重新编译,只要动了Makefile文件就得重新编译。
3.测试
./darknet detect cfg/yolov3-voc.cfg yolov3.weights data/dog.jpg
测试结果:

不显示预测图片,也不进行类别预测
解决方案:
不显示预测图片(maybe这个叫做opencv窗口?)的原因:在进行Makefile配置时,没有打开opencv,把opencv设置为1,make clean后重新make一下就出来了
不进行种类预测也不显示预测框:
肯定是yolov3.weights出现了问题
标准Yolov3.weights的大小为248M
终于出来了正常的!!!
本文档详细介绍了在Ubuntu系统下如何安装YOLOv3,包括CUDA、CUDNN和OpenCV的预装步骤,以及在安装过程中可能遇到的Makefile配置问题和解决方法,最终成功进行目标检测测试。

2914

被折叠的 条评论
为什么被折叠?



