使用mmcv库中的DCN模块代替DCNv2
MissMIA0 opened this issue · 2 comments
MissMIA0 commented
使用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,可能直接下载放到相应的文件夹里就行了
yuan243212790 commented
为什么我用mmcv代替还是同样的cuda错误
busyyang commented
屌爆了~