An Operation for 3D Deformable Convolution in Pytorch 0.3

Realize the 2D convolution, 2D and 3D deformable convolution in Pytorch 0.3.0. Did not check the 0.4.

Support the group convolution, dilate convolution, group deformable convolution, which split the channels of the input to several splits, each use the independent offset fields.

The official implementation of 2D deformable convolution

An implementation of 2D deformable convolution in Pytorch

Requirements

Pytorch = 0.3.0

Usage

Change the path in make.sh to your own path (The path of torch). Compile the op by:

bash make.sh

The .so is in xx_xx_op dir.

Use the test_function.py to test the results of the forward and the backward function.

Use the test_gradient to check the gradient of the backward. NOTE that only the double-precision version can pass the gradient check, i.e. deform2d_double. The double version is slower, so is just used for checking.

Use the test_model to check the whole model.