2020-09-08

Deformable Convolutions V2  使用技巧

1.资料源

论文名称：Deformable convnets v2: More deformable, better results

论文地址：https://arxiv.org/abs/1811.11168

代码地址（pytorch_1.0.0 branch）：https://github.com/chengdazhi/Deformable-Convolution-V2-PyTorch/tree/pytorch_1.0.0

2.如何使用

首先要确保自己的pytorch版本是正确的，然后进行C语言的编译，不然会报错，例如我用的虚拟环境rcancell，先创建screen再激活虚拟环境

No module named 'DCN'

进入pytorch的虚拟环境之后，切换到该项目的目录下，执行代码

python setup.py develop

执行完之后，C语言部分的代码会被编译好，该项目给的方法是sh make.sh，运行发现有问题（会报错ImportError: No module named torch），但是执行python setup.py develop是可以的，执行完之后在使用变形卷积的时候可能会遇到一个错误

ImportError: libcudart.so.10.0: cannot open shared object file: No such file or directory

解决方法（linux）

sudo ldconfig /usr/local/cuda-10.0/lib64

接下来就可以使用或修改该项目里面的变形卷积了，在使用中可能出现导入包出错，比如该项目下modules文件夹（包）里面deform_conv.py中的一行代码

from functions.deform_conv_func import DeformConvFunction

如果出现不能导入的问题，可将其修改为

from model.defconv.functions.deform_conv_func import DeformConvFunction

其中defconv为修改后的项目名，model为上一级目录名，需要根据自己的文件组织方式进行更改

作者Douzaikongcheng

本人博客

http://blog.csdn.net/Douzaikongcheng

qq 973912428

转载说明：请注明来源，附带本人博客链接，谢谢配合。