xingyizhou/CenterNet

使用mmcv库中的DCN模块代替DCNv2

MissMIA0 opened this issue · 2 comments

使用mmcv库中的DCN模块代替DCNv2官方库
使用 mim 安装(推荐)
mim 是 OpenMMLab 项目的包管理工具,使用它可以很方便地安装 mmcv-full。
pip install -U openmim
mim install mmcv-full

测试:
https://zhuanlan.zhihu.com/p/409033869
使用方法与官方DCNv2一样,只不过deformable_groups参数名改为deform_groups即可,
from mmcv.ops import DeformConv2dPack as DCN
dconv2 = DCN(in_channel, out_channel, kernel_size=(3, 3), stride=(2, 2), padding=1, deform_groups=2)

然后就是要下载一个权重出现urllib.error.HTTPError: HTTP Error 404: Not Found,可能直接下载放到相应的文件夹里就行了

为什么我用mmcv代替还是同样的cuda错误

屌爆了~