CenterNet icon indicating copy to clipboard operation
CenterNet copied to clipboard

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

Open MissMIA0 opened this issue 3 years ago • 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,可能直接下载放到相应的文件夹里就行了

MissMIA0 avatar Nov 11 '22 09:11 MissMIA0

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

yuan243212790 avatar Mar 01 '23 12:03 yuan243212790

屌爆了~

busyyang avatar Jul 25 '23 06:07 busyyang