中文

好的基础抵得上众多标签:标签高效的全景分割

机器人学 2024-12-04 v2 计算机视觉与模式识别

摘要

基于学习的模型在机器人感知中广泛应用的一个关键挑战是,在实现准确预测的同时显著减少所需的标注训练数据量。这不仅对于降低运营成本至关重要,而且有助于加快部署时间。在这项工作中,我们通过利用视觉基础模型奠定的基础,解决了少标签全景分割(PASTEL)的挑战。我们利用来自此类模型的描述性图像特征,训练两个轻量级网络头,分别用于语义分割和物体边界检测,仅使用极少的标注训练样本。然后,我们通过一种新颖的融合模块,基于归一化割,将它们的预测合并生成全景图。为了进一步提升性能,我们利用基于特征驱动相似性方案选择的未标注图像进行自训练。我们通过将PASTEL应用于自动驾驶和农业机器人中的重要机器人感知用例,强调了我们方法的相关性。在大量实验中,我们证明PASTEL在标签高效分割方面显著优于先前的方法,即使使用更少的标注。我们的代码公开于http://pastel.cs.uni-freiburg.de。

关键词

引用

@article{arxiv.2405.19035,
  title  = {A Good Foundation is Worth Many Labels: Label-Efficient Panoptic Segmentation},
  author = {Niclas Vödisch and Kürsat Petek and Markus Käppeler and Abhinav Valada and Wolfram Burgard},
  journal= {arXiv preprint arXiv:2405.19035},
  year   = {2024}
}