3 个月前

面向高质量候选框的类别感知动态标签分配

面向高质量候选框的类别感知动态标签分配

摘要

航空影像中的目标通常嵌入于复杂背景中,且具有任意方向。在采用定向边界框(Oriented Bounding Box, OBB)表示任意方向目标时,角度的周期性可能导致标签回归值在边界处出现不连续,从而引发损失函数的剧烈波动。为解决这一问题,本文提出一种基于复平面的OBB表示方法,并设计了一种三角函数形式的损失函数。此外,结合航空影像中复杂背景环境的先验知识以及大尺度目标间显著差异的特点,构建了一种基于Conformer结构的区域提议网络(RPN)头,用于精确预测角度信息。所提出的损失函数与Conformer RPN头协同工作,生成高质量的定向候选区域。为进一步克服仅依赖交并比(IoU)进行候选框标签分配的局限性,本文提出一种基于预测类别反馈的类别感知动态标签分配策略,使负样本选择更具代表性,有效保障分类特征与回归特征之间的一致性。在四个真实场景的定向检测数据集上进行了实验验证,结果表明,该方法在仅需极少参数调优和计算开销的情况下,显著提升了定向目标检测性能。具体而言,在DOTA-v1.0、DOTA-v1.5、DIOR-R和HRSC2016数据集上,分别取得了82.02%、71.99%、69.87%和98.77%的平均精度均值(mAP)表现。

基准测试

基准方法指标
object-detection-in-aerial-images-on-dota-1CDLA-HOP
mAP: 82.02%
object-detection-in-aerial-images-on-hrsc2016CDLA-HOP
mAP-07: 90.89
mAP-12: 98.77

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
面向高质量候选框的类别感知动态标签分配 | 论文 | HyperAI超神经