Jan, 2021

用高斯瓦瑟斯坦距离重新思考旋转物体检测

TL;DR本文提出了一种基于高斯Wasserstein距离的新型回归损失函数作为解决旋转检测回归损失设计中边界不连续性和其对最终检测度量的不一致性的基本方法,通过将旋转包围盒转换为2D高斯分布来实现,具有高斯Wasserstein距离(GWD)可以通过梯度反向传播高效地学习的优势。GWD可以在没有重叠的情况下学习,即使对于小物体检测也经常如此。GWD具有三个独特的属性,可以优雅地解决边界不连续性和类方形问题,无论包围盒如何定义。在使用不同检测器的五个数据集上的实验证明了我们方法的有效性。