ubutnu20安装CenterNet

原创已于 2022-04-23 20:53:12 修改 · 1.8k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#目标检测

于 2022-04-23 18:10:57 首次发布

该博客详细介绍了如何在Ubuntu20.04上配置Python3.6、PyTorch1.7.0+CUDA11.0的环境，并安装CenterNet及其依赖DCNv2。步骤包括安装必要的系统依赖、创建conda环境、安装PyTorch相关库、下载并编译COCOAPI和CenterNet源码，以及修改CenterNet的配置以进行KITTI数据集的测试。

部署运行你感兴趣的模型镜像

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

安装

依赖

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

新建环境

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

下载安装cocoapi：

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

下载CenterNet；
下载DCNv2_pytorch_1.x支持版本：链接1 | 链接2，命名为DCNv2，替换掉CenterNet/src/lib/models/networks/DCNv2。
更改CenterNet/src/lib/datasets/dataset/kitti.py中’…/data/kitti/training/label_val’，将其中label_val改为label_2

安装、编译

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