360BEV / README.md
insailab's picture
docs: add 360BEV model card
4f1c692 verified
|
Raw History Blame Contribute Delete
6.02 kB
---
license: apache-2.0
library_name: pytorch
tags:
- panoramic
- bird-eye-view
- semantic-mapping
- indoor-scenes
- vision
- wacv-2024
- segformer
- segnext
- 360mapper
pipeline_tag: image-segmentation
---
# 360Mapper: Panoramic Semantic Mapping for Indoor Bird's-Eye View
[IEEE/CVF WACV 2024] Official Model Checkpoints Repository for **360BEV** and **360Mapper**.
- **GitHub Repository:** https://github.com/InSAI-Lab/360BEV
- **Project Page:** https://jamycheung.github.io/360BEV.html
- **Dataset Repository:** [InSAI-Lab/360BEV](https://huggingface.co/datasets/InSAI-Lab/360BEV)
- **Paper:** [arXiv:2303.11910](https://arxiv.org/abs/2303.11910)
- **Collection:** [360BEV & 360Mapper (WACV 2024)](https://huggingface.co/collections/InSAI-Lab/360bev-and-360mapper-wacv-2024-6ad1bdcdeb1718eb4753c447)
---
## Model Description
**360Mapper** is designed for end-to-end panoramic semantic mapping to indoor Bird's-Eye View (BEV). It bridges 360Β° panoramic observations with top-down allocentric metric maps using cross-attention mechanisms, deformable attention, and high-capacity backbones (SegFormer MiT and SegNeXt MSCAN).
## Benchmark Performance
### 1. 360FV Stanford-2D3D (Front-View Semantic Segmentation)
| Model | Backbone | Input | mIoU (%) | Checkpoint |
| :--- | :--- | :---: | :---: | :--- |
| Trans4PASS | MiT-B2 | RGB | 52.10 | -- |
| CBFC | ResNet-101 | RGB | 52.20 | -- |
| **Ours (360FV)** | **MiT-B2** | **RGB** | **54.30** | [`360FV_Stanford/`](https://huggingface.co/InSAI-Lab/360BEV/tree/main/360FV_Stanford) |
### 2. 360FV Matterport3D (Front-View Semantic Segmentation)
| Model | Backbone | Input | mIoU (%) | Checkpoint |
| :--- | :--- | :---: | :---: | :--- |
| HoHoNet | ResNet-101 | RGB-D | 44.85 | -- |
| SegFormer | MiT-B2 | RGB | 45.53 | -- |
| **Ours (360FV)** | **MiT-B2** | **RGB** | **46.35** | [`360FV_Matterport/fv_mp3d.pkl`](https://huggingface.co/InSAI-Lab/360BEV/resolve/main/360FV_Matterport/fv_mp3d.pkl) |
### 3. 360BEV-Stanford (Panoramic BEV Semantic Mapping)
| Method | Backbone | Acc (%) | mRecall (%) | mPrecision (%) | mIoU (%) | Checkpoint |
| :--- | :---: | :---: | :---: | :---: | :---: | :--- |
| Trans4Map | MiT-B0 | 86.41 | 40.45 | 57.47 | 32.26 | -- |
| Trans4Map | MiT-B2 | 86.53 | 45.28 | 62.61 | 36.08 | -- |
| **360Mapper (Ours)** | **MiT-B0** | 92.07 | 50.14 | 65.37 | 42.42 | [`360BEV_Stanford/360BEV_s2d3d_segformer_b0.pkl`](https://huggingface.co/InSAI-Lab/360BEV/resolve/main/360BEV_Stanford/360BEV_s2d3d_segformer_b0.pkl) |
| **360Mapper (Ours)** | **MiT-B2** | **92.80** | 53.56 | 67.72 | 45.78 | [`360BEV_Stanford/360BEV_s2d3d_segformer_b2.pkl`](https://huggingface.co/InSAI-Lab/360BEV/resolve/main/360BEV_Stanford/360BEV_s2d3d_segformer_b2.pkl) |
| **360Mapper (Ours)** | **MSCA-B** | 92.67 | **55.02** | **68.02** | **46.44** | [`360BEV_Stanford/360BEV_s2d3d_segnext_B.pkl`](https://huggingface.co/InSAI-Lab/360BEV/resolve/main/360BEV_Stanford/360BEV_s2d3d_segnext_B.pkl) |
### 4. 360BEV-Matterport (Panoramic BEV Semantic Mapping)
| Method | Backbone | Acc (%) | mRecall (%) | mPrecision (%) | mIoU (%) | Checkpoint |
| :--- | :---: | :---: | :---: | :---: | :---: | :--- |
| Trans4Map | MiT-B0 | 70.19 | 44.31 | 50.39 | 31.92 | -- |
| Trans4Map | MiT-B2 | 73.28 | 51.60 | 53.02 | 36.72 | -- |
| **360Mapper (Ours)** | **MiT-B0** | 75.44 | 48.80 | 56.01 | 36.98 | [`360BEV_Matterport/360BEV_mp3d_segformer_b0.pkl`](https://huggingface.co/InSAI-Lab/360BEV/resolve/main/360BEV_Matterport/360BEV_mp3d_segformer_b0.pkl) |
| **360Mapper (Ours)** | **MiT-B2** | 78.80 | 59.54 | 59.97 | 44.32 | [`360BEV_Matterport/360BEV_mp3d_segformer_b2.pkl`](https://huggingface.co/InSAI-Lab/360BEV/resolve/main/360BEV_Matterport/360BEV_mp3d_segformer_b2.pkl) |
| **360Mapper (Ours)** | **MSCA-B** | **78.93** | **60.51** | **62.83** | **46.31** | [`360BEV_Matterport/360BEV_mp3d_segnext_B.pkl`](https://huggingface.co/InSAI-Lab/360BEV/resolve/main/360BEV_Matterport/360BEV_mp3d_segnext_B.pkl) |
## Checkpoints Organization
```text
β”œβ”€β”€ 360FV_Stanford/
β”‚ β”œβ”€β”€ fv_s2d3d_fold1.pkl
β”‚ β”œβ”€β”€ fv_s2d3d_fold2.pkl
β”‚ └── fv_s2d3d_fold3.pkl
β”œβ”€β”€ 360FV_Matterport/
β”‚ └── fv_mp3d.pkl
β”œβ”€β”€ 360BEV_Stanford/
β”‚ β”œβ”€β”€ 360BEV_s2d3d_segformer_b0.pkl
β”‚ β”œβ”€β”€ 360BEV_s2d3d_segformer_b2.pkl
β”‚ └── 360BEV_s2d3d_segnext_B.pkl
└── 360BEV_Matterport/
β”œβ”€β”€ 360BEV_mp3d_segformer_b0.pkl
β”œβ”€β”€ 360BEV_mp3d_segformer_b2.pkl
└── 360BEV_mp3d_segnext_B.pkl
```
## Quick Start & Evaluation
### 1. Download Weights
Using Hugging Face CLI:
```bash
# Download all checkpoints
hf download InSAI-Lab/360BEV --local-dir ./checkpoints
# Or download specific model
hf download InSAI-Lab/360BEV 360BEV_Stanford/360BEV_s2d3d_segformer_b2.pkl --local-dir ./checkpoints
```
Using Python:
```python
from huggingface_hub import hf_hub_download
model_path = hf_hub_download(
repo_id="InSAI-Lab/360BEV",
filename="360BEV_Stanford/360BEV_s2d3d_segformer_b2.pkl",
local_dir="./checkpoints"
)
```
### 2. Evaluation
In the [360BEV codebase](https://github.com/InSAI-Lab/360BEV):
```bash
# 360BEV Matterport
python test_360BEV_Matterport.py --config configs/model_360BEV_mp3d.yml
# 360BEV Stanford-2D3D
python test_360BEV_S2d3d.py --config configs/model_360BEV_s2d3d.yml
# Stanford2D3D (360FV)
python test_pano_360Attention_S2d3d.py --config configs/model_fv_s2d3d.yml
# 360FV-Matterport
python test_pano_360Attention_Matterport.py --config configs/model_fv_mp3d.yml
```
## Citation
```bibtex
@inproceedings{teng2024_360bev,
title={360BEV: Panoramic Semantic Mapping for Indoor Bird's-Eye View},
author={Teng, Zhifeng and Zhang, Jiaming and Yang, Kailun and Peng, Kunyu and Shi, Hao and Reiß, Simon and Cao, Ke and Stiefelhagen, Rainer},
booktitle={2024 IEEE/CVF Winter Conference on Applications of Computer Vision (WACV)},
year={2024}
}
```
## License
This repository is licensed under the [Apache-2.0 License](https://www.apache.org/licenses/LICENSE-2.0).