DCNv2

Redes convolucionais deformáveis v2 com 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 

Observação

Agora a filial principal é para Pytorch 1.x, você pode voltar para Pytorch 0.4 com,

git checkout pytorch_0.4

Questões conhecidas:

• Verificação de gradiente WRT Offset (resolvido)
• Para trás não é reentrante (menor)

Esta é uma adaptação das redes deformáveis oficiais.

ATUALIZAÇÃO: Todos os passes de verificação de gradiente com precisão dupla .

Outra questão é que ele levanta RuntimeError: Backward is not reentrant . No entanto, o erro é muito pequeno ( <1e-7 para flutuação <1e-15 para o dobro), por isso pode não ser um problema sério (?)

Por favor, poste um problema ou relações públicas se tiver algum comentário.