GitHub - matroid/DCNv2 at 455dd87363fa337c7e17a9f1a2ee72e66db87c1b

matroid / DCNv2 Public

forked from CharlesShang/DCNv2

Notifications You must be signed in to change notification settings
Fork 0
Star 0

Deformable Convolutional Networks v2 with Pytorch

BSD-3-Clause license

0 stars 402 forks Branches Tags Activity

Star

Notifications

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 34 Commits
src		src
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
__init__.py		__init__.py
dcn_v2.py		dcn_v2.py
make.sh		make.sh
setup.py		setup.py
testcpu.py		testcpu.py
testcuda.py		testcuda.py

Repository files navigation

Deformable Convolutional Networks V2 with Pytorch 1.0

CPU and GPU version

Build

    ./make.sh         # build
    python testcpu.py    # run examples and gradient check on cpu
    python testcuda.py   # run examples and gradient check on gpu

This repository contains the Pytorch DCNv2 implementation by Charles Shang. It has been modified so that it can run on both CPU and GPU.