安装CenterNet环境

0、首先创建文件夹。

  a、mkdir detect

  b、cd detect

1、安装conda。

  a、./Anaconda3-2019.10-Linux-x86_64.sh

2、创建并激活CenterNet环境。

  a、conda create --name CenterNet python=3.7

  b、conda activate CenterNet

3、安装pytorch。

  a、conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

4、安装CocoAPI。

  a、git clone https://github.com/cocodataset/cocoapi.git

  b、cd PythonAPI

  c、make

    出现错误：“gcc: error: pycocotools/_mask.c: No such file or directory”

    解决方法：pip install cython

  d、python setup.py install --user

5、安装CenterNet和其他依赖。

  a、cd ../..

  b、git clone https://github.com/xingyizhou/CenterNet

  c、cd CenterNet

  d、pip install -r requirements.txt

6、编译DCNV2。

  a、cd src/lib/models/networks/DCNv2

  b、./make.sh

    出现错误：“ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead.”

    解决方法：cd ..

                      mv DCNv2/ DCNv2_old/

                      git clone https://github.com/CharlesShang/DCNv2.git

                      cd DCNv2

                      ./make.sh

7、编译NMS。

  a、cd ../../../external

  b、make