Home

Awesome

CoBEVT: Cooperative Bird's Eye View Semantic Segmentation with Sparse Transformers [CORL2022]

paper supplement video

This is the official implementation of CoRL2022 paper "CoBEVT: Cooperative Bird's Eye View Semantic Segmentation with Sparse Transformers". Runsheng Xu, Zhengzhong Tu, Hao Xiang, Wei Shao, Bolei Zhou, Jiaqi Ma

UCLA, UT-Austin

<br> <div align="center"><img src="images/CorpBEVT_Overview-1.png" width="85%"/></div> <div align="center"> <b>Overview of CoBEVT</b> </div> <br>

Introduction

CoBEVT is the first generic multi-agent multi-camera perception framework that can cooperatively generate BEV map predictions. The core component of CoBEVT, named fused axial attention or FAX module, can capture sparsely local and global spatial interactions across views and agents. We achieve SOTA performance both on OPV2V and nuScenes dataset with real-time performance.

<br> <div align="center"><img src="images/nuscene.gif" width="75%"/></div> <div align="center"> <b>nuScenes demo:</b> Our CoBEVT can be used on single-vehicle multi-camera semantic BEV Segmentations. </div> <br> <br> <div align="center"><img src="images/opv2v.gif" width="75%"/></div> <div align="center"> <b>OPV2V demo:</b> Our CoBEVT can also be used for multi-agent BEV map prediction. </div> <br>

Installation

The pipeline for nuScenes dataset and OPV2V dataset is different. Please refer to the specific folder for more details based on your research purpose.

:point_right: nuScenes Users <br/> :point_right: OPV2V Users

Models

<details> <summary><strong>Fused Axial Attention Module (FAX)</strong> (click to expand) </summary> <img src = "https://user-images.githubusercontent.com/43280278/192188069-44381995-7d0f-43fb-aded-68d62595b2d4.png" width="800"> </details> <details> <summary><strong>SinBEVT (single-agent multi-view fusion) and FuseBEVT (multi-agent BEV fusion) </strong> (click to expand) </summary> <img src = "https://user-images.githubusercontent.com/43280278/192188116-2b3fd013-b8fc-4d79-a5dd-5eb88d09e622.png" width="800"> </details> <details> <summary><strong>CoBEVT Architecture</strong> (click to expand) </summary> <img src = "images/CorpBEVT_Overview-1.png" width="800"> </details>

Results

<details> <summary><strong>Main results (OPV2V-camera, -LiDAR, and nuScenes.)</strong> (click to expand) </summary> <img src = "https://user-images.githubusercontent.com/43280278/192186838-8b42605b-9cb0-4f3e-9a44-615ec158ce37.png" width="800"> </details> <details> <summary><strong>Qualitative results on OPV2V-camera</strong> (click to expand) </summary> <img src = "https://user-images.githubusercontent.com/43280278/192187203-4716e5dc-af4d-4652-bddb-fb28ab07260d.png" width="1000"> <img src = "https://user-images.githubusercontent.com/43280278/192187720-a1eb5c39-5a71-48c8-8eff-6b04b158d0d4.png" width="1000"> </details> <details> <summary><strong>Qualitative results on OPV2V-LiDAR</strong> (click to expand) </summary> <img src = "https://user-images.githubusercontent.com/43280278/192187375-0a7168bb-8be9-49f6-9f2b-42b7b57fe031.png" width="1000"> <img src = "https://user-images.githubusercontent.com/43280278/192187420-8c8c7b31-ee09-4d9f-8df2-7e92dd79acb2.png" width="1000"> </details> <details> <summary><strong>Qualitative results on nuScenes</strong> (click to expand) </summary> <img src = "https://user-images.githubusercontent.com/43280278/192187578-1127a7ae-590e-4f27-bcc2-311fd554d0ae.png" width="1000"> </details> <details> <summary><strong>Ablation study</strong> (click to expand) </summary> <img src = "https://user-images.githubusercontent.com/43280278/192186995-0fa0b1dd-b5a8-4125-af39-17a79ce0de0e.png" width="800"> </details>

Citation

@inproceedings{xu2022cobevt,
 author = {Runsheng Xu, Zhengzhong Tu, Hao Xiang, Wei Shao, Bolei Zhou, Jiaqi Ma},
 title = {CoBEVT: Cooperative Bird's Eye View Semantic Segmentation with Sparse Transformers},
 booktitle={Conference on Robot Learning (CoRL)},
 year = {2022}}
@article{xu2022v2x,
 title={V2X-ViT: Vehicle-to-everything cooperative perception with vision transformer},
 author={Xu, Runsheng and Xiang, Hao and Tu, Zhengzhong and Xia, Xin and Yang, Ming-Hsuan and Ma, Jiaqi},
 journal={Proceedings of the European Conference on Computer Vision (ECCV)},
 year={2022}
}
@inproceedings{tu2022maxim,
 title={Maxim: Multi-axis mlp for image processing},
 author={Tu, Zhengzhong and Talebi, Hossein and Zhang, Han and Yang, Feng and Milanfar, Peyman and Bovik, Alan and Li, Yinxiao},
 booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
 pages={5769--5780},
 year={2022}
}
@article{tu2022maxvit,
 title={Maxvit: Multi-axis vision transformer},
 author={Tu, Zhengzhong and Talebi, Hossein and Zhang, Han and Yang, Feng and Milanfar, Peyman and Bovik, Alan and Li, Yinxiao},
 journal={Proceedings of the European Conference on Computer Vision (ECCV)},
 year={2022}
}

Acknowledgement

CoBEVT is build upon OpenCOOD, which is the first Open Cooperative Detection framework for autonomous driving.

Our nuScenes experiments used the training pipeline in CVT(CVPR2022).

CoBEVT is partly inspired by V2X-ViT, MAXIM and MaxViT.