yolov3图像检测CPU版和GPU版环境搭建

1 yolov3搭建CPU环境

1.0 软件环境及插件

• Mac OS 10.13.2
• Homebrew
• wget

1.2 安装步骤

【(1) 安装Homebrew】

/usr/bin/ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"

【(2) 安装wget】

brew install wget

【(3) 获取图像检测训练模型】

git clone https://github.com/pjreddie/darknet

【(4) 编译】

cd darknet
make

【(5) 获取训练模型权重】

wget https://pjreddie.com/media/files/yolov3.weights

【(6) 测试】

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

【(7) 结果】

图1.0 CPU处理结果

【(8) CPU处理耗时】

图1.2 CPU处理耗时

---

2 yolov3搭建GPU环境

2.1 软件环境及插件

• Ubuntu 16.04.5
• wget

2.2 安装步骤

【(1) 获取图像检测训练模型】

git clone https://github.com/pjreddie/darknet

【(2) GPU参数配置】

# 进入darknet目录
cd darknet
# 打开Makefile文件
vim Makefile

【(3) 修改Makefile文件】

# 使用GPU，设为1
GPU=1
# 启用CUDNN，设为1
CUDNN=1

【(4) 编译】

make
make clean(清除编译)

【(5) 获取训练模型权重】

wget https://pjreddie.com/media/files/yolov3.weights

【(6) 测试】

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

【(7) 结果】

图2.1 GPU处理结果

【(8) GPU处理耗时】

图2.2 GPU处理耗时

---

[参考文献]
[1]https://pjreddie.com/darknent/yolo/
[2]https://prjreddie.com/darknet/install/#cuda

---