CenterNet
CenterNet copied to clipboard
使用mmcv库中的DCN模块代替DCNv2
使用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错误
屌爆了~