DCNv2

Redes convolucionales v2 deformables con Pytorch 1.x

Construir

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

Nota

Ahora la rama maestra es para Pytorch 1.x, puede volver a Pytorch 0.4 con,

git checkout pytorch_0.4

Problemas conocidos:

• CHECK de gradiente WRT Offset (resuelto)
• Hacia atrás no es reentrante (menor)

Esta es una adaptación de los connets de conformidad deformables oficiales.

ACTUALIZACIÓN: Todos los pases de verificación de gradiente con doble precisión.

Otro problema es que plantea RuntimeError: Backward is not reentrant . Sin embargo, el error es muy pequeño ( <1e-7 para flotación <1e-15 para el doble), por lo que puede no ser un problema grave (?)

Publique un problema o PR si tiene algún comentario.