DCNv2

Deformable Convolutional Networks V2 dengan Pytorch 1.x

Membangun

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

Catatan

Sekarang cabang utama adalah untuk Pytorch 1.x, Anda dapat beralih kembali ke Pytorch 0.4 dengan,

git checkout pytorch_0.4

Masalah yang Diketahui:

• Gradien periksa wrt offset (diselesaikan)
• Backward tidak masuk kembali (minor)

Ini adalah adaptasi dari konvnet resmi yang dapat dideformasi.

UPDATE: Semua tiket pemeriksaan gradien dengan presisi ganda .

Masalah lain adalah meningkatkan RuntimeError: Backward is not reentrant . Namun, kesalahannya sangat kecil ( <1e-7 untuk float <1e-15 untuk ganda), jadi mungkin bukan masalah serius (?)

Silakan posting masalah atau PR jika Anda memiliki komentar.