Mar, 2024

遥感图像中基于显著对比学习的少样本目标检测

TL;DR通过使用定向边界框而不是传统的水平边界框来学习更好的特征表示,针对远程感知图像,我们提出了一种新的用于远程感知图像的少样本定向目标检测方法,称为记忆对比学习的少样本定向目标检测(FOMC)。我们引入了一个带有动态更新存储器的监督对比学习模块,以应对对象误分类的具有挑战性的问题。该模块可以利用大批量的负样本,并增强模型学习未见类别的区分特征的能力。在DOTA和HRSC2016数据集上进行了全面实验,我们的模型在少样本定向目标检测任务上实现了最先进的性能。发布代码和预训练模型。