arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2308.14525cs.CV

视觉鸟瞰图语义分割的半监督学习

Semi-Supervised Learning for Visual Bird's Eye View Semantic Segmentation

Institute of Cyber-Systems and Control, Zhejiang University(浙江大学 Cyber-Systems 与控制研究所);China Mobile Research Institute(中国移动研究院)

Junyu Zhu, Lina Liu, Yu Tang, Feng Wen, Wanlong Li, Yong Liu

AI总结:

本文提出首个面向视觉BEV语义分割的半监督框架,通过一致性损失和保持几何关系的联合旋转数据增强利用无标签数据,在nuScenes和Argoverse数据集上有效提升了预测精度。

摘要:

视觉鸟瞰图(BEV)语义分割帮助自动驾驶车辆仅从图像中理解周围环境,包括静态元素(如道路)和动态元素(如车辆、行人)。然而,全监督方法标注过程的高成本限制了视觉BEV语义分割的能力,其通常需要高精地图、3D目标边界框和相机外参矩阵。在本文中,我们提出了一种新颖的视觉BEV语义分割半监督框架,通过在训练过程中利用无标签图像来提升性能。随后提出了一种充分利用无标签数据的一致性损失,不仅在语义预测上约束模型,还在BEV特征上约束模型。此外,我们提出了一种新颖且有效的数据增强方法,名为联合旋转(conjoint rotation),该方法在保持前视图像与BEV语义分割之间几何关系的同时合理地增强数据集。在nuScenes和Argoverse数据集上的大量实验表明,我们的半监督框架能够有效提高预测精度。据我们所知,这是首个探索利用无标签数据提升视觉BEV语义分割性能的工作。代码可在https://github.com/Junyu-Z/Semi-BEVseg获取。

英文摘要:

Visual bird's eye view (BEV) semantic segmentation helps autonomous vehicles understand the surrounding environment only from images, including static elements (e.g., roads) and dynamic elements (e.g., vehicles, pedestrians). However, the high cost of annotation procedures of full-supervised methods limits the capability of the visual BEV semantic segmentation, which usually needs HD maps, 3D object bounding boxes, and camera extrinsic matrixes. In this paper, we present a novel semi-supervised framework for visual BEV semantic segmentation to boost performance by exploiting unlabeled images during the training. A consistency loss that makes full use of unlabeled data is then proposed to constrain the model on not only semantic prediction but also the BEV feature. Furthermore, we propose a novel and effective data augmentation method named conjoint rotation which reasonably augments the dataset while maintaining the geometric relationship between the front-view images and the BEV semantic segmentation. Extensive experiments on the nuScenes and Argoverse datasets show that our semi-supervised framework can effectively improve prediction accuracy. To the best of our knowledge, this is the first work that explores improving visual BEV semantic segmentation performance using unlabeled data. The code is available at https://github.com/Junyu-Z/Semi-BEVseg

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑