CenterNet 编译DCNv2踩坑记录

一直卡在DCNv2的编译上,尝试根据报错解决和查阅网上的各种教程均告失败,最后看到有人说DCNv2的版本要和torch版本对应

GitHub - lbin/DCNv2 at pytorch_1.6

把网址后面的版本改成自己的版本,下载后更换掉原先的DCNv2文件夹 就可以正常编译了

如果实在编译不过,也可以选择不使用添加了DCN的骨干网络,这样就不需要通过DCN的编译了。修改src/lib/models/model.py,将12、13行:

from .networks.pose_dla_dcn import get_pose_net as get_dla_dcn
from .networks.resnet_dcn import get_pose_net as get_pose_net_dcn

删去

第17行开始:

_model_factory = {
  'res': get_pose_net, # default Resnet with deconv
  'dlav0': get_dlav0, # default DLAup
  'dla': get_dla_dcn,
  'resdcn': get_pose_net_dcn,
  'hourglass': get_large_hourglass_net,
}

中对应的带有dcn的骨干网络键值对删去,即删去:

'dla': get_dla_dcn,
'resdcn': get_pose_net_dcn,

 这样就不会报错了。

你可能感兴趣的:(学习记录,pytorch,深度学习)