DCNv2

Réseaux de convolution déformables V2 avec pytorch 1.x

Construire

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

Note

Maintenant, la branche maître est pour Pytorch 1.x, vous pouvez revenir à Pytorch 0.4 avec,

git checkout pytorch_0.4

Problèmes connus:

• Contrôle du gradient WRT Offset (résolu)
• En arrière n'est pas réentrant (mineur)

Il s'agit d'une adaptation des connonnets déformables officiels.

MISE À JOUR: Tous les vérifications du gradient passent avec double précision.

Un autre problème est qu'il augmente RuntimeError: Backward is not reentrant . Cependant, l'erreur est très petite ( <1e-7 pour le flotteur <1e-15 pour le double), donc ce n'est peut-être pas un problème grave (?)

Veuillez publier un problème ou un PR si vous avez des commentaires.