DCNv2

Verformbare Faltungsnetze V2 mit Pytorch 1.x

Bauen

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

Notiz

Jetzt ist der Master -Zweig für Pytorch 1.x.

git checkout pytorch_0.4

Bekannte Probleme:

• Gradient Check WRT -Offset (gelöst)
• Rückwärts ist nicht wieder eingetragen (minderjährig)

Dies ist eine Anpassung der offiziellen deformierbaren Konvazen.

UPDATE: Alle Gradientenüberprüfungen werden mit doppelter Präzision erfolgen.

Ein weiteres Problem ist, dass es RuntimeError: Backward is not reentrant . Der Fehler ist jedoch sehr klein ( <1e-7 für Float <1e-15 für doppelt), daher ist es möglicherweise kein ernstes Problem (?)

Bitte veröffentlichen Sie ein Problem oder eine PR, wenn Sie Kommentare haben.