1zb/deformable-convolution-pytorch

PyTorch implementation of Deformable Convolution

Cuda

Deformable Convolutional Networks in PyTorch

This repo is an implementation of Deformable Convolution. Ported from author's MXNet implementation.

Build

sh make.sh
CC=g++ python build.py

See test.py for example usage.

Notice

Only torch.cuda.FloatTensor is supported.