docs: add Trans4PASS model card
Browse files
README.md
ADDED
|
@@ -0,0 +1,103 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
library_name: pytorch
|
| 4 |
+
tags:
|
| 5 |
+
- panoramic
|
| 6 |
+
- vision-transformer
|
| 7 |
+
- semantic-segmentation
|
| 8 |
+
- domain-adaptation
|
| 9 |
+
- trans4pass
|
| 10 |
+
- trans4pass-plus
|
| 11 |
+
- cvpr-2022
|
| 12 |
+
pipeline_tag: image-segmentation
|
| 13 |
+
---
|
| 14 |
+
|
| 15 |
+
# Trans4PASS & Trans4PASS+: Transformers for Panoramic Semantic Segmentation
|
| 16 |
+
|
| 17 |
+
Official Model Checkpoints Repository for **Trans4PASS** and **Trans4PASS+**.
|
| 18 |
+
|
| 19 |
+
- **GitHub Repository:** https://github.com/InSAI-Lab/Trans4PASS
|
| 20 |
+
- **Paper (Trans4PASS, CVPR 2022):** [arXiv:2203.01452](https://arxiv.org/abs/2203.01452)
|
| 21 |
+
- **Paper (Trans4PASS+, arXiv 2022):** [arXiv:2207.11860](https://arxiv.org/abs/2207.11860)
|
| 22 |
+
- **Dataset Repository:** [InSAI-Lab/SynPASS](https://huggingface.co/datasets/InSAI-Lab/SynPASS)
|
| 23 |
+
|
| 24 |
+
---
|
| 25 |
+
|
| 26 |
+
## Checkpoints Organization
|
| 27 |
+
|
| 28 |
+
This repository contains all pretrained backbones, supervised weights, and domain adaptation checkpoints matching the Trans4PASS project directory structure:
|
| 29 |
+
|
| 30 |
+
```text
|
| 31 |
+
βββ pretrained/
|
| 32 |
+
β βββ mit_b1.pth
|
| 33 |
+
β βββ mit_b2.pth
|
| 34 |
+
βββ adaptations/
|
| 35 |
+
β βββ snapshots/
|
| 36 |
+
β βββ CS2DensePASS_Trans4PASS_v1_MPA/BestCS2DensePASS_G.pth
|
| 37 |
+
β βββ CS2DensePASS_Trans4PASS_v2_MPA/BestCS2DensePASS_G.pth
|
| 38 |
+
β βββ CS132CS132DP13_Trans4PASS_plus_v2_MPA/BestCS132DP13_G.pth
|
| 39 |
+
β βββ CS2DP_Trans4PASS_plus_v1_MPA/BestCS2DensePASS_G.pth
|
| 40 |
+
β βββ CS2DP_Trans4PASS_plus_v2_MPA/BestCS2DensePASS_G.pth
|
| 41 |
+
βββ workdirs/
|
| 42 |
+
βββ cityscapes/ (trans4pass & plus, tiny & small)
|
| 43 |
+
βββ cityscapes13/ (trans4pass & plus, tiny & small)
|
| 44 |
+
βββ stanford2d3d/ (trans4pass & plus, tiny & small)
|
| 45 |
+
βββ stanford2d3d8/ (trans4pass & plus, tiny & small)
|
| 46 |
+
βββ stanford2d3d_pan/ (folds F1, F2, F3)
|
| 47 |
+
βββ structured3d8/ (trans4pass & plus, tiny & small)
|
| 48 |
+
βββ synpass/ (trans4pass & plus, tiny & small)
|
| 49 |
+
βββ synpass13/ (trans4pass & plus, tiny & small)
|
| 50 |
+
```
|
| 51 |
+
|
| 52 |
+
## Benchmark Results
|
| 53 |
+
|
| 54 |
+
### Cityscapes -> DensePASS (Domain Adaptation)
|
| 55 |
+
| Model | CS mIoU (%) | DP mIoU (%) |
|
| 56 |
+
| :--- | :---: | :---: |
|
| 57 |
+
| Trans4PASS (Tiny) | 72.49 | 45.89 |
|
| 58 |
+
| Trans4PASS (Small) | 72.84 | 51.38 |
|
| 59 |
+
| Trans4PASS+ (Tiny) | 72.67 | 50.23 |
|
| 60 |
+
| Trans4PASS+ (Small) | **75.24** | **51.41** |
|
| 61 |
+
|
| 62 |
+
### SynPASS Benchmark
|
| 63 |
+
| Model | Val mIoU (%) | Test mIoU (%) |
|
| 64 |
+
| :--- | :---: | :---: |
|
| 65 |
+
| Trans4PASS (Tiny) | 43.68 | 38.53 |
|
| 66 |
+
| Trans4PASS (Small) | 44.80 | 38.57 |
|
| 67 |
+
| Trans4PASS+ (Tiny) | 45.21 | 38.85 |
|
| 68 |
+
| Trans4PASS+ (Small) | **46.47** | **39.16** |
|
| 69 |
+
|
| 70 |
+
## Download & Usage
|
| 71 |
+
|
| 72 |
+
Using `hf`:
|
| 73 |
+
|
| 74 |
+
```bash
|
| 75 |
+
# Download all models to the local directory
|
| 76 |
+
hf download InSAI-Lab/Trans4PASS --local-dir .
|
| 77 |
+
```
|
| 78 |
+
|
| 79 |
+
Evaluation example:
|
| 80 |
+
|
| 81 |
+
```bash
|
| 82 |
+
# In Trans4PASS repository
|
| 83 |
+
python tools/eval_sp.py --config-file configs/synpass/trans4pass_plus_tiny_512x512.yaml
|
| 84 |
+
```
|
| 85 |
+
|
| 86 |
+
## Citation
|
| 87 |
+
|
| 88 |
+
```bibtex
|
| 89 |
+
@inproceedings{zhang2022bending,
|
| 90 |
+
title={Bending Reality: Distortion-aware Transformers for Adapting to Panoramic Semantic Segmentation},
|
| 91 |
+
author={Zhang, Jiaming and Yang, Kailun and Ma, Chaoxiang and Rei{\ss}, Simon and Peng, Kunyu and Stiefelhagen, Rainer},
|
| 92 |
+
booktitle={CVPR},
|
| 93 |
+
pages={16917--16927},
|
| 94 |
+
year={2022}
|
| 95 |
+
}
|
| 96 |
+
|
| 97 |
+
@article{zhang2022behind,
|
| 98 |
+
title={Behind Every Domain There is a Shift: Adapting Distortion-aware Vision Transformers for Panoramic Semantic Segmentation},
|
| 99 |
+
author={Zhang, Jiaming and Yang, Kailun and Shi, Hao and Rei{\ss}, Simon and Peng, Kunyu and Ma, Chaoxiang and Fu, Haodong and Wang, Kaiwei and Stiefelhagen, Rainer},
|
| 100 |
+
journal={arXiv preprint arXiv:2207.11860},
|
| 101 |
+
year={2022}
|
| 102 |
+
}
|
| 103 |
+
```
|