最近跑centerNet的模型,但是在DCN部分遇到错误,记录一下
配置:ubuntu18.04
cuda10.0
pytorch:1.10
网络: https://github.com/xingyizhou/CenterNet
问题1:编译自带的DCNv2的时候./make.sh
ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead.
https://github.com/CharlesShang/DCNv2
这个问题网上问的也比较多
最后我的解决方式为pytorch1.1.0变为pytorch1.3.1模型可以成功运行DCN的网络部分.
在make以后测试的时候遇到以下输出,不用管它,运行centerNet的网络,可以正常运行
版本:
torch 1.3.1
torchvision 0.4.2