Dec, 2023

面向定向物体检测的边缘瓦瑟斯坦距离损失

TL;DR回归损失设计是面向对象检测的一个重要课题,由于角度的周期性和宽度和高度定义的不确定性,传统的L1距离损失及其变种一直以来都面临着度量不连续和类似于方形的问题,分布基方法则通过将有方向的框表示为分布而显示出显著优势,我们提出了一种新颖的有方向回归损失,Wasserstein距离(EWD)损失,以缓解类似于方形的问题。同时,基于有方向框的边缘表示,EWD损失可以推广到四边形和多项式回归场景,对多个常用数据集和不同的检测器的实验表明了该方法的有效性。