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
- 修改Makefile文件
- 编译
- 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