中文

MapTR:面向在线矢量化高精地图构建的结构化建模与学习

计算机视觉与模式识别 2023-01-31 v2 机器人学

摘要

高精(HD)地图提供驾驶场景丰富而精确的环境信息,是自动驾驶系统中规划的基础且不可或缺的组件。我们提出 MapTR,一种用于高效在线矢量化高精地图构建的结构化端到端 Transformer。我们提出统一的置换等价建模方法,即将地图元素建模为具有一组等价置换的点集,从而准确描述地图元素的形状并稳定学习过程。我们设计了一种分层查询嵌入方案,以灵活编码结构化地图信息,并执行分层二分匹配用于地图元素学习。在 nuScenes 数据集上,仅以相机输入,MapTR 在现有矢量化地图构建方法中取得了最佳性能与效率。特别地,MapTR-nano 在 RTX 3090 上以实时推理速度(25.125.1 FPS)运行,比现有最先进的基于相机的方法快 8×8\times,同时 mAP 高出 5.05.0。即便与现有最先进的多模态方法相比,MapTR-nano 的 mAP 高出 0.70.7,而 MapTR-tiny 的 mAP 高出 13.513.5 且推理速度快 3×3\times。大量定性结果表明,MapTR 在复杂多样的驾驶场景中保持稳定鲁棒的地图构建质量。MapTR 在自动驾驶中具有重大应用价值。代码与更多演示见 \url{https://github.com/hustvl/MapTR}。

关键词

引用

@article{arxiv.2208.14437,
  title  = {MapTR: Structured Modeling and Learning for Online Vectorized HD Map Construction},
  author = {Bencheng Liao and Shaoyu Chen and Xinggang Wang and Tianheng Cheng and Qian Zhang and Wenyu Liu and Chang Huang},
  journal= {arXiv preprint arXiv:2208.14437},
  year   = {2023}
}

备注

Accepted to ICLR 2023 as Spotlight Presentation. Code&demos: https://github.com/hustvl/MapTR