成功在RTX3080上训练centernet

成功在RTX3080上训练centernet

1、源码下载地址
Centernet
or https://github.com/xingyizhou/CenterNet
2、环境配置教程地址【需要做些改动】
centernet installment
https://github.com/xingyizhou/CenterNet/blob/master/readme/INSTALL.md

1）步骤1中在官网下载和RTX3080符合的pytorch，亲测下面的版本可行。

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

2）按照步骤安装COCOAPI和requirement.txt,即步骤2和4。
3）编译 DCNv2需要在下面网址下载新版的 DCNv2，否则会编译失败。将新下载的文件夹替换CenterNet/src/lib/models/networks中的DCNv2.

https://codechina.csdn.net/mirrors/jinfagang/DCNv2_latest

在DCNv2中执行

python setup.py install --user

此时环境就配置好了
3、如果想要跑自己的数据集，可以参考如下网址，步骤很详细
训练自己的数据集步骤