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

1 yolov3搭建CPU环境

1.1 软件环境及插件

• Mac OS 10.13.2
• Homebrew
• wget

1.2 安装步骤

• 安装Homebrew
  /usr/bin/ruby -e “$(curl -fsSL
  https://raw.githubusercontent.com/Homebrew/install/master/install)”
• 安装wget
  brew install wget
• 获取图像检测训练模型
  git clone https://github.com/pjreddie/darknet
• 编译
  cd darknet
  make
• 获取训练模型权重
  wget https://pjreddie.com/media/files/yolov3.weights
• 测试
  ./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
• 结果
  
• CPU处理耗时
  

---

2 yolov3搭建GPU环境

2.1 软件环境及插件

• Ubuntu 16.04.5
• wget

2.2 安装步骤

• 获取图像检测训练模型
  git clone https://github.com/pjreddie/darknet
• GPU参数配置
  
  • 修改Makefile文件
    
    • cd darknet
    • vim Makefile
    • GPU=1
    • CUDNN=1
• 编译
  
  • make
  • make clean(清除编译)
• 获取训练模型权重
  wget https://pjreddie.com/media/files/yolov3.weights
• 测试
  ./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
• 结果
  
• GPU处理耗时
  

---

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

---