DCNv2

可變形的捲積網絡v2具有Pytorch 1.x

建造

    ./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偏移（已解決）
• 向後不是再入（次要）

這是對官方可變形convnets的改編。

更新：所有梯度檢查通過均具有雙重精度。

另一個問題是，它增加了RuntimeError: Backward is not reentrant 。但是，錯誤非常小（float <1e-7 <1e-15 ，double）可能不是一個嚴重的問題（？）

如果您有任何評論，請發布問題或公關。