Project Icon

BEVFormer

多摄像头鸟瞰图学习框架助力自动驾驶感知

BEVFormer是一个用于自动驾驶感知的开源框架,通过时空Transformer从多摄像头图像中学习统一的鸟瞰图表示。该方法利用预定义的网格查询,结合空间交叉注意力和时间自注意力机制,有效聚合多视角的空间和时序信息。在nuScenes测试集上,BEVFormer达到56.9%的NDS指标,显著超越现有方法,与激光雷达系统性能相当。这一创新为基于纯视觉的3D目标检测提供了新的基准。

BEVFormer:基于摄像头的检测的前沿基线

https://user-images.githubusercontent.com/27915819/161392594-fc0082f7-5c37-4919-830a-2dd423c1d025.mp4

BEVFormer:通过时空变换器从多摄像头图像学习鸟瞰图表示,ECCV 2022

新闻

  • [2022/6/16]:我们新增了两个BEVformer配置,比基础版本需要更少的GPU内存。请拉取此仓库以获取最新代码。
  • [2022/6/13]:我们发布了BEVFormer的初始版本。它在nuScenes上实现了51.7%的NDS基线结果。
  • [2022/5/23]:🚀🚀基于BEVFormer构建的BEVFormer++,集合了近期最佳实践和我们独特的修改,在Waymo开放数据集3D仅摄像头检测挑战赛中排名第一。我们将在CVPR 2022自动驾驶研讨会上展示BEVFormer++。
  • [2022/3/10]:🚀BEVFormer在nuScenes检测任务上以56.9% NDS(仅使用摄像头)达到了最新技术水平!

摘要

在这项工作中,作者提出了一个名为BEVFormer的新框架,它通过时空变换器学习统一的BEV表示,以支持多个自动驾驶感知任务。简而言之,BEVFormer通过预定义的网格形BEV查询与空间和时间空间进行交互,从而利用空间和时间信息。为了聚合空间信息,作者设计了一种空间交叉注意力机制,每个BEV查询从跨摄像头视图的感兴趣区域提取空间特征。对于时间信息,作者提出了一种时间自注意力机制,以递归方式融合历史BEV信息。 该方法在nuScenes测试集上实现了新的最先进的56.9%的NDS指标,比之前的最佳技术高出9.0个百分点,并与基于LiDAR的基线性能相当。

方法

方法

入门指南

模型库

骨干网络方法学习率调度NDSmAP内存配置下载
R50BEVFormer-tiny_fp1624轮35.925.7-配置模型/日志
R50BEVFormer-tiny24轮35.425.26500M配置模型/日志
R101-DCNBEVFormer-small24轮47.937.010500M配置模型/日志
R101-DCNBEVFormer-base24轮51.741.628500M配置模型/日志
R50BEVformerV2-t1-base24轮42.635.123952M配置模型/日志
R50BEVformerV2-t1-base48轮43.935.923952M配置模型/日志
R50BEVformerV2-t124轮45.338.137579M配置模型/日志
R50BEVformerV2-t148轮46.539.537579M配置模型/日志
R50BEVformerV2-t224轮51.842.038954M配置模型/日志
R50BEVformerV2-t248轮52.643.138954M配置模型/日志
R50BEVformerV2-t824轮55.346.040392M配置模型/日志
BEVFormerV2模型和日志的百度网盘链接[https://pan.baidu.com/s/1ynzlAt1DQbH8NkqmisatTw?pwd=fdcv]在这里。

目录

  • BEVFormerV2 超查询
  • BEVFormerV2 优化,包括内存、速度、推理
  • BEVFormerV2 发布
  • BEV分割检查点
  • BEV分割代码
  • 3D检测检查点
  • 3D检测代码
  • 初始化

引用

如果本工作对您的研究有帮助,请考虑引用以下BibTeX条目。

@article{li2022bevformer,
  title={BEVFormer: Learning Bird's-Eye-View Representation from Multi-Camera Images via Spatiotemporal Transformers},
  author={Li, Zhiqi and Wang, Wenhai and Li, Hongyang and Xie, Enze and Sima, Chonghao and Lu, Tong and Qiao, Yu and Dai, Jifeng}
  journal={arXiv preprint arXiv:2203.17270},
  year={2022}
}
@article{Yang2022BEVFormerVA,
  title={BEVFormer v2: Adapting Modern Image Backbones to Bird's-Eye-View Recognition via Perspective Supervision},
  author={Chenyu Yang and Yuntao Chen and Haofei Tian and Chenxin Tao and Xizhou Zhu and Zhaoxiang Zhang and Gao Huang and Hongyang Li and Y. Qiao and Lewei Lu and Jie Zhou and Jifeng Dai},
  journal={ArXiv},
  year={2022},
}

致谢

非常感谢这些优秀的开源项目:

↳ 点赞者

Stargazers repo roster for @nastyox/Repo-Roster

↳ 分叉者

Forkers repo roster for @nastyox/Repo-Roster

项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型,免费访问各种图像生成与AI训练工具,从Stable Diffusion等基础模型开始,轻松实现创新图像生成。体验前沿的AI技术,推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器,它将改变您观看视频的方式!SubCat结合了先进的人工智能技术,为您提供即时视频字幕翻译,无论是本地视频还是网络流媒体,让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台,利用前沿AI技术,为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图,再到文案生成,美间让创意设计更简单、更高效。

Project Cover

AIWritePaper论文写作

AIWritePaper论文写作是一站式AI论文写作辅助工具,简化了选题、文献检索至论文撰写的整个过程。通过简单设定,平台可快速生成高质量论文大纲和全文,配合图表、参考文献等一应俱全,同时提供开题报告和答辩PPT等增值服务,保障数据安全,有效提升写作效率和论文质量。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号