【深度学习】Darknet框架在Ubuntu16.04上的安装和测试

Darknet主页

Darknet: Open Source Neural Networks in C

安装过程

1. clone源码到本地

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

2. 修改Makefile

如果有GPU，并且安装了cuda

GPU=1

如果安装了cudnn

CUDNN=1

如果源码编译了OpenCV

OPENCV=1

不要忘记根据显卡和cuda版本修改ARCH

3. 编译安装

make

4. 添加环境变量

export PATH="/thirdparty-source/darknet":$PATH
export C_INCLUDE_PATH="/thirdparty-source/darknet/include":$C_INCLUDE_PATH
export CPLUS_INCLUDE_PATH="/thirdparty-source/darknet/include":$CPLUS_INCLUDE_PATH
export LD_LIBRARY_PATH="/thirdparty-source/darknet":$LD_LIBRARY_PATH
export LIBRARY_PATH="/thirdparty-source/darknet":$LIBRARY_PATH

5. 测试

darknet imtest /thirdparty-source/darknet/data/eagle.jpg

6. 使用预训练的yolov3权重文件检测图片

下载预训练的yolov3权重文件

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

选择GPU编号0运行

./darknet -i 0 detect cfg/yolov3.cfg /home/zhangchen/yolov3.weights data/dog.jpg

选择CPU运行

./darknet -nogpu detect cfg/yolov3.cfg /home/zhangchen/yolov3.weights data/dog.jpg

注意即使添加了环境变量PATH，这里也必须在darknet文件夹下运行darknet可执行文件，否则会报路径相关的错误

结语

如果您有修改意见或问题，欢迎留言或者通过邮箱和我联系。
手打很辛苦，如果我的文章对您有帮助，转载请注明出处。