当前位置：网站首页>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://yzsam.com/2022/04/202204232048354054.html

边栏推荐

猜你喜欢

随机推荐