CN.md

December 15, 2023 · View on GitHub

English | 简体中文

Towards Robust Bird's Eye View Perception under Common Corruption and Domain Shift

Shaoyuan Xie1   Lingdong Kong2,3   Wenwei Zhang2,4   Jiawei Ren4   Liang Pan4   Kai Chen2   Ziwei Liu4
1华中科技大学   2上海人工智能实验室   3新加坡国立大学   4南洋理工大学S-Lab

项目概览

RoboBEV 是首个为在自然数据"损坏"和域迁移条件下, 基于相机的鸟瞰图 (BEV) 感知量身定制的鲁棒性评估基线。该基线包括了以下八种可能出现在驾驶场景中的数据"损坏"类型: 1传感器故障损坏、2运动和数据处理损坏、3光照条件损坏和4天气条件损坏。

左前视角前视角右前视角左前视角前视角右前视角
左后视角后视角右后视角左后视角后视角右后视角

请参阅我们的 项目主页 以获取更多细节与实例。 :blue_car:

版本更新

  • [2023.06] - nuScenes-C 数据集现已发布在OpenDataLab平台!🚀
  • [2023.04] - 我们在 Paper-with-Code 平台搭建了 "鲁棒BEV感知" 基线。现在就加入鲁棒性评测吧!:raising_hand:
  • [2023.02] - 我们邀请每一位BEV爱好者参与到 "鲁棒BEV感知" 基线中来! 更多细节,请阅读此页面。:beers:
  • [2023.01] - 推出 "RoboBEV"! 在这个初始版本中,11个BEV检测算法和1个单目3D检测算法已经在8个"损坏"类型和3种严重程度下进行了基准测试。

大纲

安装

请参阅 安装.md 以获取更多有关环境安装的细节。

数据准备

我们的数据集由 OpenDataLab 平台搭载。


OpenDataLab 是一个引领AI大模型时代的数据开源开放平台。OpenDataLab 为人工智能研究者提供免费开源的数据集,通过该平台,研究者可以获得格式统一的各领域经典数据集。

请参阅 数据准备.md 以获取更多有关准备 nuScenesnuScenes-C 数据集的细节。

开始实验

请参阅 开始实验.md 以获取更多有关如何使用本代码库的细节。

模型库

 基于多视角相机的BEV检测模型
 基于单目相机的3D物体检测模型
 基于相机与激光雷达融合的BEV检测模型
 基于多视角相机的BEV图分割模型
 基于多视角相机的深度估计模型
 基于多视角相机的语义占用模型

鲁棒性基线

:triangular_ruler: 指标: 在我们的基准中,nuScenes Detection Score (NDS) 被用作评价模型性能的主要指标。我们采用以下两个指标来比较模型的鲁棒性:

  • mCE (越低越好): 候选模型的平均损坏误差 (百分比),这是在三种严重程度的所有"损坏"类型中与基线模型相比计算出来的。
  • mRR (越高越好): 候选模型的平均复原率 (百分比) ,这是在三种严重程度的所有"损坏"类型中与它的"干净"性能相比计算出来的。

:gear: 注释: 符号 :star: 表示 mCE 计算中采用的基线模型。更详细的实验结果,请参考 实验结果.md.

BEV 检测

模型mCE (%) \downarrowmRR (%) \uparrowCleanCam CrashFrame LostColor QuantMotion BlurBrightLow LightFogSnow
DETR3D:star:100.0070.770.42240.28590.26040.31770.26610.40020.27860.39120.1913
DETR3DCBGS99.2170.020.43410.29910.26850.32350.25420.41540.27660.40200.1925
BEVFormerSmall101.2359.070.47870.27710.24590.32750.25700.37410.24130.35830.1809
BEVFormerBase97.9760.400.51740.31540.30170.35090.26950.41840.25150.40690.1857
PETRR50-p4111.0161.260.36650.23200.21660.24720.22990.28410.15710.28760.1417
PETRVoV-p4100.6965.030.45500.29240.27920.29680.24900.38580.23050.37030.2632
ORA3D99.1768.630.44360.30550.27500.33600.26470.40750.26130.39590.1898
BEVDetR50115.1251.830.37700.24860.19240.24080.20610.25650.11020.24610.0625
BEVDetR101113.6853.120.38770.26220.20650.25460.22650.25540.11180.24950.0810
BEVDetR101-pt112.8056.350.37800.24420.19620.30410.25900.25990.13980.20730.0939
BEVDetSwinT116.4846.260.40370.26090.21150.22780.21280.21910.04900.24500.0680
BEVDepthR50110.0256.820.40580.26380.21410.27510.25130.28790.17570.29030.0863
BEVerseSwinT110.6748.600.46650.31810.30370.26000.26470.26560.05930.27810.0644
BEVerseSwinS117.8249.570.49510.33640.24850.28070.26320.33940.11180.28490.0985
PolarFormerR10196.0670.880.46020.31330.28080.35090.32210.43040.25540.42620.2304
PolarFormerVoV98.7567.510.45580.31350.28110.30760.23440.42800.24410.40610.2468
SRCN3DR10199.6770.230.42860.29470.26810.33180.26090.40740.25900.39400.1920
SRCN3DVoV102.0467.950.42050.28750.25790.28270.21430.38860.22740.37740.2499
Sparse4DR101100.0155.040.54380.28730.26110.33100.25140.39840.25100.38840.2259
SOLOFusionshort108.6861.450.39070.25410.21950.28040.26030.29660.20330.29980.1066
SOLOFusionlong97.9964.420.48500.31590.24900.35980.34600.40020.28140.39910.1480
SOLOFusionfusion92.8664.530.53810.38060.34640.40580.36420.43290.26260.44800.1376
FCOS3Dfinetune107.8262.090.39490.28490.24790.25740.25700.32180.14680.33210.1136
BEVFusionCam109.0257.810.41210.27770.22550.27630.27880.29020.10760.30410.1461
BEVFusionLiDAR--0.6928--------
BEVFusionC+L43.8097.410.71380.69630.69310.70440.69770.70180.6787--
TransFusion--0.68870.68430.64470.68190.67490.68430.6663--
AutoAlignV2--0.61390.58490.58320.60060.59010.60760.5770--

多视角相机的深度估计

ModelMetricCleanCam CrashFrame LostColor QuantMotion BlurBrightLow LightFogSnow
SurroundDepthAbs Rel0.2800.4850.4970.3340.3380.3390.3540.3200.423

多视角相机的语义占用

ModelMetricCleanCam CrashFrame LostColor QuantMotion BlurBrightLow LightFogSnow
TPVFormermIoU vox52.0627.3922.8538.1638.6449.0037.3846.6919.39
SurroundOccSC mIoU20.3011.6010.0014.0312.4119.1812.1518.427.39

BEV模型标定

模型预训练时序建模深度估计CBGS骨干网络BEV编码器图像尺寸mCE (%)mRR (%)NDS
DETR3DResNetAttention1600×900100.0070.770.4224
DETR3DCBGSResNetAttention1600×90099.2170.020.4341
BEVFormerSmallResNetAttention1280×720101.2359.070.4787
BEVFormerBaseResNetAttention1600×90097.9760.400.5174
PETRR50-p4ResNetAttention1408×512111.0161.260.3665
PETRVoV-p4VoVNetV2Attention1600×900100.6965.030.4550
ORA3DResNetAttention1600×90099.1768.630.4436
PolarFormerR101ResNetAttention1600×90096.0670.880.4602
PolarFormerVoVVoVNetV2Attention1600×90098.7567.510.4558
SRCN3DR101ResNetCNN+Attn.1600×90099.6770.230.4286
SRCN3DVoVVoVNetV2CNN+Attn.1600×900102.0467.950.4205
Sparse4DR101ResNetCNN+Attn.1600×900100.0155.040.5438
BEVDetR50ResNetCNN704×256115.1251.830.3770
BEVDetR101ResNetCNN704×256113.6853.120.3877
BEVDetR101-ptResNetCNN704×256112.8056.350.3780
BEVDetSwinTSwinCNN704×256116.4846.260.4037
BEVDepthR50ResNetCNN704×256110.0256.820.4058
BEVerseSwinTSwinCNN704×256137.2528.240.1603
BEVerseSwinTSwinCNN704×256110.6748.600.4665
BEVerseSwinSSwinCNN1408×512132.1329.540.2682
BEVerseSwinSSwinCNN1408×512117.8249.570.4951
SOLOFusionshortResNetCNN704×256108.6861.450.3907
SOLOFusionlongResNetCNN704×25697.9964.420.4850
SOLOFusionfusionResNetCNN704×25692.8664.530.5381

注: 预训练表示从FCOS3D初始化的模型。时序建模表示是否使用了时间信息。深度估计表示具有显式深度估计分支的模型。CBGS表示模型使用类平衡的分组采样策略。

生成"损坏"数据

你可以创建你自己的 "RoboBEV" 数据集! 请参考文件:数据生成.md.

更新计划

  • 初始更新已放出. 🚀
  • 新增生成"损坏"数据的运行脚本.
  • 新增nuScenes-C数据集下载链接.
  • 新增模型评测的运行脚本.
  • 新增BEV地图分割模型.
  • 新增多视角深度估计模型.
  • 新增多视角语义分割模型.
  • ...

引用

如果你认为这项工作对你有帮助,请考虑引用以下内容:

@article{xie2023robobev,
    title = {RoboBEV: Towards Robust Bird's Eye View Perception under Corruptions},
    author = {Xie, Shaoyuan and Kong, Lingdong and Zhang, Wenwei and Ren, Jiawei and Pan, Liang and Chen, Kai and Liu, Ziwei},
    journal = {arXiv preprint arXiv:2304.06719}, 
    year = {2023}
}
@misc{xie2023robobev_codebase,
    title = {The RoboBEV Benchmark for Robust Bird's Eye View Detection under Common Corruption and Domain Shift},
    author = {Xie, Shaoyuan and Kong, Lingdong and Zhang, Wenwei and Ren, Jiawei and Pan, Liang and Chen, Kai and Liu, Ziwei},
    howpublished = {\url{https://github.com/Daniel-xsy/RoboBEV}},
    year = {2023}
}

许可

Creative Commons License
这项工作是在 Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License 下进行的。这个代码库中的一些模型可能是采用其他许可证。如果你将我们的代码用于商业用途, 请参考 许可.md 以进行更仔细的检查。

致谢

这项工作是基于 MMDetection3D 代码库.


MMDetection3D 是一个基于PyTorch的开源目标检测工具箱,面向下一代通用三维检测平台。它是由MMLab开发的OpenMMLab项目的一部分。

:heart: 我们感谢 Jiangmiao Pang 和 Tai Wang 的建设性的讨论和反馈,感谢 OpenDataLab 平台托管我们的数据集。