用于快速卫星目标检测的软标签

图像分类中的软标签是图像真实分类的向量表示。在本文中，我们研究卫星目标检测背景下的软标签。我们提出以检测结果为基来构建一个新的软标签数据集。创建高质量模型的大量精力在于收集和标注训练数据。如果我们能用模型为我们生成数据集，我们不仅能快速创建数据集，还能补充现有的开源数据集。使用 xView 数据集的一个子集，我们训练了一个 YOLOv5 模型来检测汽车、飞机和船只。然后我们用该模型为第二个训练集生成软标签，再据此训练并与原始模型比较。我们表明软标签可用于训练一个几乎与在原始数据上训练的模型一样准确的模型。