File size: 6,017 Bytes
4f1c692 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 | ---
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).
|