dcn和dcnv2 torch

原文:

https://blog.csdn.net/justsolow/article/details/105971437

这个有个问题,训练时,设置不同的gpu时,默认绑定0显卡,导致报错

1.用torch自带的DeformConv2d

from torchvision.ops import DeformConv2d

需要前传时需要自己创建offset。

2.改进版:

DCN和DCNv2(可变性卷积)
网上关于两篇文章的详细描述已经很多了,我这里具体的细节就不多讲了,只说一下其中实现起来比较困惑的点。(黑体字会讲解)

DCNv1解决的问题就是我们常规的图像增强,仿射变换(线性变换加平移)不能解决的多种形式目标变换的几何变换的问题。如下图所示。

可变性卷积的思想很简单,就是讲原来固定形状的卷积核变成可变的。如下图所示:

猜你喜欢

转载自blog.csdn.net/jacke121/article/details/114941347