原文:
https://blog.csdn.net/justsolow/article/details/105971437
这个有个问题,训练时,设置不同的gpu时,默认绑定0显卡,导致报错
1.用torch自带的DeformConv2d
from torchvision.ops import DeformConv2d
需要前传时需要自己创建offset。
2.改进版:
DCN和DCNv2(可变性卷积)
网上关于两篇文章的详细描述已经很多了,我这里具体的细节就不多讲了,只说一下其中实现起来比较困惑的点。(黑体字会讲解)
DCNv1解决的问题就是我们常规的图像增强,仿射变换(线性变换加平移)不能解决的多种形式目标变换的几何变换的问题。如下图所示。
可变性卷积的思想很简单,就是讲原来固定形状的卷积核变成可变的。如下图所示: