YOLOv3 安装步骤

Posted on | In | views
  • 操作系统：Linux 16.04
  • 依赖：
    • CUDA
    • OpenCV
    • darknet

写在前面：conda 是一款非常好用的 python 环境管理工具，建议安装 Anaconda 或 Miniconda。安装及使用请参阅网上教程，安装完成后记得添加清华大学 tuna 镜像。

Anaconda 镜像使用帮助

CUDA

按照网上教程或英伟达官方网站正确安装 CUDA 和 cuDNN（实验室服务器上的版本是 cuda9.0 cudnn7.1）

OpenCv

如果安装了 conda，首先进入你想要安装 YOLO 的虚拟环境。第一次使用 Anaconda 请先创建虚拟环境。

1
conda create --name YOURNAME python=3.6 numpy pandas matplotlib

（YOURNAME 替换为你想要的名字）上述命令将创建新的 python 虚拟环境，并安装常用工具包 numpy, pandas, matplotlib，创建完成后进入虚拟环境

1
2
source activate YOURNAME
source deactivate    # use this one to exit

进入环境后开始安装 OpenCV

1
conda install opencv

安装完毕即可

darknet

首先下载 github 仓库

1
2
git clone https://github.com/AlexeyAB/darknet.git
cd darknet

然后打开 Makefile 文件，修改以下选项：

1
2
3
GPU = 1
CUDNN = 1
OPENCV = 1

然后在 darknet 目录下执行

1
make

如果没有报错，YOLOv3 就安装成功了。测试一下

1
./darknet

可以看到输出为：

1
usage: ./darknet <function>

Reference

  1. https://github.com/AlexeyAB/darknet
  2. https://pjreddie.com/darknet/yolo/ 🦄