insailab commited on
Commit
8fb1903
Β·
verified Β·
1 Parent(s): ae5792b

docs: add CMNeXt model card

Browse files
Files changed (1) hide show
  1. README.md +127 -0
README.md ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ library_name: pytorch
4
+ tags:
5
+ - multimodal
6
+ - semantic-segmentation
7
+ - vision
8
+ - cvpr-2023
9
+ - cmnext
10
+ - segformer
11
+ - swin-transformer
12
+ pipeline_tag: image-segmentation
13
+ ---
14
+
15
+ # CMNeXt: Delivering Arbitrary-Modal Semantic Segmentation
16
+
17
+ [CVPR 2023] Official Model Checkpoints Repository for **CMNeXt**.
18
+
19
+ - **Project Page:** https://jamycheung.github.io/DELIVER.html
20
+ - **GitHub Repository:** https://github.com/InSAI-Lab/DELIVER
21
+ - **Paper:** [arXiv:2303.01480](https://arxiv.org/abs/2303.01480)
22
+ - **Dataset Repository:** [InSAI-Lab/DELIVER](https://huggingface.co/datasets/InSAI-Lab/DELIVER)
23
+
24
+ ---
25
+
26
+ ## Model Description
27
+
28
+ **CMNeXt** is an arbitrary cross-modal semantic segmentation model operating in the Hub2Fuse paradigm with asymmetric branches:
29
+ - Multi-Head Self-Attention (MHSA) blocks in the RGB branch
30
+ - Parallel Pooling Mixer (PPX) blocks in the accompanying modality branch
31
+ - Self-Query Hub selects informative supplementary features
32
+ - Feature Rectification Module (FRM) and Feature Fusion Module (FFM) fuse features dynamically across 1 to 81 modalities
33
+
34
+ ## Checkpoints Organization
35
+
36
+ This repository contains trained weights and pretrained backbones matching the official DELIVER codebase structure:
37
+
38
+ ```text
39
+ β”œβ”€β”€ DELIVER/
40
+ β”‚ β”œβ”€β”€ cmnext_b2_deliver_rgb.pth
41
+ β”‚ β”œβ”€β”€ cmnext_b2_deliver_rgbd.pth
42
+ β”‚ β”œβ”€β”€ cmnext_b2_deliver_rgbde.pth
43
+ β”‚ β”œβ”€β”€ cmnext_b2_deliver_rgbdel.pth
44
+ β”‚ β”œβ”€β”€ cmnext_b2_deliver_rgbdl.pth
45
+ β”‚ β”œβ”€β”€ cmnext_b2_deliver_rgbe.pth
46
+ β”‚ └── cmnext_b2_deliver_rgbl.pth
47
+ β”œβ”€β”€ KITTI360/
48
+ β”‚ β”œβ”€β”€ cmnext_b2_kitti360_rgb.pth
49
+ β”‚ β”œβ”€β”€ cmnext_b2_kitti360_rgbd.pth
50
+ β”‚ β”œβ”€β”€ cmnext_b2_kitti360_rgbde.pth
51
+ β”‚ β”œβ”€β”€ cmnext_b2_kitti360_rgbdel.pth
52
+ β”‚ β”œβ”€β”€ cmnext_b2_kitti360_rgbdl.pth
53
+ β”‚ β”œβ”€β”€ cmnext_b2_kitti360_rgbe.pth
54
+ β”‚ └── cmnext_b2_kitti360_rgbl.pth
55
+ β”œβ”€β”€ MCubeS/
56
+ β”‚ β”œβ”€β”€ cmnext_b2_mcubes_rgb.pth
57
+ β”‚ β”œβ”€β”€ cmnext_b2_mcubes_rgba.pth
58
+ β”‚ β”œβ”€β”€ cmnext_b2_mcubes_rgbad.pth
59
+ β”‚ └── cmnext_b2_mcubes_rgbadn.pth
60
+ β”œβ”€β”€ MFNet/
61
+ β”‚ └── cmnext_b4_mfnet_rgbt.pth
62
+ β”œβ”€β”€ NYU_Depth_V2/
63
+ β”‚ └── cmnext_b4_nyu_rgbd.pth
64
+ β”œβ”€β”€ UrbanLF/
65
+ β”‚ β”œβ”€β”€ cmnext_b4_urbanlf_real_rgblf1.pth
66
+ β”‚ β”œβ”€β”€ cmnext_b4_urbanlf_real_rgblf33.pth
67
+ β”‚ β”œβ”€β”€ cmnext_b4_urbanlf_real_rgblf8.pth
68
+ β”‚ β”œβ”€β”€ cmnext_b4_urbanlf_real_rgblf80.pth
69
+ β”‚ β”œβ”€β”€ cmnext_b4_urbanlf_syn_rgblf1.pth
70
+ β”‚ β”œβ”€β”€ cmnext_b4_urbanlf_syn_rgblf33.pth
71
+ β”‚ β”œβ”€β”€ cmnext_b4_urbanlf_syn_rgblf8.pth
72
+ β”‚ └── cmnext_b4_urbanlf_syn_rgblf80.pth
73
+ └── pretrained/
74
+ β”œβ”€β”€ segformers/
75
+ β”‚ β”œβ”€β”€ mit_b0.pth ~ mit_b5.pth
76
+ └── swintransformer/
77
+ β”œβ”€β”€ swin_base_patch4_window12_384_22k.pth
78
+ β”œβ”€β”€ swin_large_patch4_window12_384_22k.pth
79
+ └── swin_small_patch4_window7_224.pth
80
+ ```
81
+
82
+ ## Benchmark Results
83
+
84
+ ### DELIVER Benchmark
85
+ | Model-Modal | #Params(M) | GFLOPs | mIoU (%) | Checkpoint |
86
+ | :--- | :--- | :--- | :--- | :--- |
87
+ | CMNeXt-RGB | 25.79 | 38.93 | 57.20 | `DELIVER/cmnext_b2_deliver_rgb.pth` |
88
+ | CMNeXt-RGB-E | 58.69 | 62.94 | 57.48 | `DELIVER/cmnext_b2_deliver_rgbe.pth` |
89
+ | CMNeXt-RGB-L | 58.69 | 62.94 | 58.04 | `DELIVER/cmnext_b2_deliver_rgbl.pth` |
90
+ | CMNeXt-RGB-D | 58.69 | 62.94 | 63.58 | `DELIVER/cmnext_b2_deliver_rgbd.pth` |
91
+ | CMNeXt-RGB-D-E | 58.72 | 64.19 | 64.44 | `DELIVER/cmnext_b2_deliver_rgbde.pth` |
92
+ | CMNeXt-RGB-D-L | 58.72 | 64.19 | 65.50 | `DELIVER/cmnext_b2_deliver_rgbdl.pth` |
93
+ | CMNeXt-RGB-D-E-L | 58.73 | 65.42 | **66.30** | `DELIVER/cmnext_b2_deliver_rgbdel.pth` |
94
+
95
+ ### Other Benchmarks
96
+ - **KITTI-360:** CMNeXt-RGB-D-E-L achieves 67.84% mIoU
97
+ - **NYU Depth V2:** CMNeXt-RGB-D (MiT-B4) achieves 56.90% mIoU
98
+ - **MFNet (RGB-T):** CMNeXt-RGB-T (MiT-B4) achieves 59.90% mIoU
99
+ - **UrbanLF:** Up to 83.22% mIoU (Real) / 81.02% mIoU (Synthetic)
100
+ - **MCubeS:** CMNeXt-RGB-A-D-N achieves 51.54% mIoU
101
+
102
+ ## Download & Usage
103
+
104
+ Using `hf`:
105
+
106
+ ```bash
107
+ # Download all checkpoints into output directory
108
+ hf download InSAI-Lab/CMNeXt --local-dir output/
109
+ ```
110
+
111
+ Evaluate with DELIVER repository:
112
+
113
+ ```bash
114
+ cd DELIVER
115
+ CUDA_VISIBLE_DEVICES=0 python tools/val_mm.py --cfg configs/deliver_rgbdel.yaml
116
+ ```
117
+
118
+ ## Citation
119
+
120
+ ```bibtex
121
+ @inproceedings{zhang2023delivering,
122
+ title={Delivering Arbitrary-Modal Semantic Segmentation},
123
+ author={Zhang, Jiaming and Liu, Ruiping and Shi, Hao and Yang, Kailun and Rei{\ss}, Simon and Peng, Kunyu and Fu, Haodong and Wang, Kaiwei and Stiefelhagen, Rainer},
124
+ booktitle={CVPR},
125
+ year={2023}
126
+ }
127
+ ```