jianghuairong / deformable-convolution-pytorch

PyTorch implementation of Deformable Convolution

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

About

PyTorch implementation of Deformable Convolution


Languages

Language:Cuda 41.2%Language:C 38.1%Language:Python 16.2%Language:C++ 4.2%Language:Shell 0.2%