当前位置：网站首页>ubutnu20安装CenterNet

ubutnu20安装CenterNet

2022-04-23 20:48:00 【CC977】

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

安装

新建环境

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。

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测试

版权声明
本文为[CC977]所创，转载请带上原文链接，感谢
https://blog.csdn.net/CC977/article/details/124366748

边栏推荐

猜你喜欢

随机推荐