一 CPU版本
我的系统是Ubuntu14.04,无GPU。
1.安装opencv,3.0以上的版本。请自行百度安装。
2.安装darknet.
打开YOLO主页
https://pjreddie.com/darknet/yolo/
按照步骤安装:
下载源码
git clone https://github.com/pjreddie/darknet
编译
cd darknet
make
make是若是报错
/usr/bin/ld: cannot find -lippic
则在自己电脑上找到 libippicv.a,放入 /usr/local/lib中,然后重启终端,再次make,可以解决。
下载网络参数:
wget https://pjreddie.com/media/files/yolo.weights
测试:
./darknet detect cfg/yolo.cfg yolo.weights data/dog.jpg
出现图片后则证明安装成功
二 配置GPU版本
1.打开 ./makefile,更改选项:
前提是机器上已经安装了opencv,cuda,cudnn。
GPU=1
CUDNN=1
OPENCV=1
OPENMP=0
DEBUG=0
2. 更改GPU算力配置
#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_20,code=[sm_20,sm_21] \ This one is deprecated?
# This is what I use, uncomment if you know your arch and want to specify
ARCH= -gencode arch=compute_61,code=compute_61
我的电脑配置是两张TITAN XP, 在NIVIDA官网查询算力是6.1,所以改成以上情况。
修改完后保存,重新编译,与 一 的步骤一样。