ImportError: cannot import name ‘deform_conv_cuda’ from ‘assets.ops.dcn’
在DBNet++的复现中,搭建好环境进行自己的数据集训练时,运行train.py文件,出现了报错。
这里出现问题是在DB-master\assets\ops\dcn文件夹下,出问题的主要语句是在dcn文件夹下的function中deform_conv.py和deform_pool.py里。
其中的from … import deform_conv_cuda
如下列
deform_conv_cuda不能够正常调用,我在网上找到了这个问题出现的原因。
根因是mmdetection版本问题,在mmdetction github的issue中有对应的讨论。网址如下:
https://github.com/open-mmlab/mmdetection/issues/413#issuecomment-515299000
根据其中的高赞回答,我将解决方法锁定在
在DBNet模型训练的具体实现中,这要求我们cd到DB-master\assets\ops\dcn文件夹下,执行操作
python setup.py develop