ubutnu20安装CenterNet

环境：Ubuntu20.04 + python3.6 + pytorch1.7.0 + cuda11.0

安装

1. 依赖

   apt-get install texlive-extra-utils
   apt-get install gnuplot
   apt-get install ghostscript
   sudo apt-get install libboost-all-dev
   

2. 新建环境

   conda create --name centernet python=3.6 
   conda activate centernet
   conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 -c pytorch
   conda install numpy cython matplotlib
   

3. 下载安装cocoapi：

   conda activate centernet
   cd $COCOAPI/PythonAPI
   make
   python setup.py install --user
   

4. 下载CenterNet；

5. 下载DCNv2_pytorch_1.x支持版本：链接1 | 链接2，命名为DCNv2，替换掉CenterNet/src/lib/models/networks/DCNv2。

6. 更改CenterNet/src/lib/datasets/dataset/kitti.py中’…/data/kitti/training/label_val’，将其中label_val改为label_2

7. 安装、编译

   conda activate centernet
   pip install -r requirements.txt
   cd CenterNet/src/lib/models/networks/DCNv2
   ./make.sh
   #安装完成，以下可选
   #务必先安装cocoapi再执行下面命令
   cd CenterNet/src/lib/external
   make
   

KITTI测试