PANet 安装步骤

Posted on | In | views

Requirements

  • 操作系统：Linux 16.04
  • 平台：PyTorch
  • 依赖
    • pytorch = 0.4.0
    • torchvision >= 0.2.0
    • cython
    • matplotlib
    • numpy
    • scipy
    • opencv
    • pyyaml
    • packaging
    • pycocotools – for coco dataset
    • tensorboardX – for logging in TensorBoard

PyTorch

使用 conda 安装 PyTorch 比较简单，首先进入 conda 环境，然后执行

1
conda install pytorch=0.4.0 torchvision cuda90 -c pytorch

其余 cython 等包都可以使用 conda install 或者 pip install 命令来安装，就不再重复了

编译 PANet

首先下载 github 仓库：

1
2
git clone https://github.com/ShuLiu1993/PANet.git
cd PANet

然后编译

1
2
cd lib  # please change to this directory
sh make.sh

等待编译完成即可。训练代码在 PANet/tools 文件夹下。

如果安装过程报错找不到 cuda_runtime.h，可以查看这里

Reference

  1. https://github.com/ShuLiu1993/PANet
  2. https://github.com/roytseng-tw/Detectron.pytorch