DCNv2

Pytorch 1.x를 사용한 변형 가능한 컨볼 루션 네트워크 v2

짓다

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

메모

이제 마스터 브랜치는 Pytorch 1.x를위한 것이므로 Pytorch 0.4로 다시 전환 할 수 있습니다.

git checkout pytorch_0.4

알려진 문제 :

• 그라디언트 점검 WRT 오프셋 (해결)
• 뒤로는 재진입이 아닙니다 (미성년자)

이것은 공식 변형 가능한 컨벤트의 적응입니다.

업데이트 : 모든 그라디언트 체크는 이중 정밀도로 통과합니다.

또 다른 문제는 RuntimeError: Backward is not reentrant . 그러나 오류는 매우 작습니다 (플로트 <1e-15 의 경우 <1e-7 ). 심각한 문제가 아닐 수도 있습니다 (?)

의견이 있으면 문제 또는 PR을 게시하십시오.