deformable-convolution-pytorch icon indicating copy to clipboard operation
deformable-convolution-pytorch copied to clipboard

PyTorch implementation of Deformable Convolution

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.