insailab commited on
Commit
5bf049c
Β·
verified Β·
1 Parent(s): b119f8f

docs: add Trans4PASS model card

Browse files
Files changed (1) hide show
  1. README.md +103 -0
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
+ ```