HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

MMSFormer: Multimodal Transformer for Material and Semantic Segmentation

Md Kaykobad Reza; Ashley Prater-Bennette; M. Salman Asif

MMSFormer: Multimodal Transformer for Material and Semantic Segmentation

Abstract

Leveraging information across diverse modalities is known to enhance performance on multimodal segmentation tasks. However, effectively fusing information from different modalities remains challenging due to the unique characteristics of each modality. In this paper, we propose a novel fusion strategy that can effectively fuse information from different modality combinations. We also propose a new model named Multi-Modal Segmentation TransFormer (MMSFormer) that incorporates the proposed fusion strategy to perform multimodal material and semantic segmentation tasks. MMSFormer outperforms current state-of-the-art models on three different datasets. As we begin with only one input modality, performance improves progressively as additional modalities are incorporated, showcasing the effectiveness of the fusion block in combining useful information from diverse input modalities. Ablation studies show that different modules in the fusion block are crucial for overall model performance. Furthermore, our ablation studies also highlight the capacity of different input modalities to improve performance in the identification of different types of materials. The code and pretrained models will be made available at https://github.com/csiplab/MMSFormer.

Code Repositories

csiplab/mmsformer
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
semantic-segmentation-on-fmb-datasetMMSFormer (RGB-Infrared)
mIoU: 61.70
semantic-segmentation-on-fmb-datasetMMSFormer (RGB)
mIoU: 57.20
semantic-segmentation-on-mcubesMMSFormer (RGB-A-D-N)
mIoU: 53.11%
semantic-segmentation-on-mcubesMMSFormer (RGB-A)
mIoU: 51.30%
semantic-segmentation-on-mcubesMMSFormer (RGB)
mIoU: 50.44%
semantic-segmentation-on-mcubesMMSFormer (RGB-A-D)
mIoU: 52.05%
semantic-segmentation-on-mcubes-pMMSFormer (RGB-A)
mIoU: 51.30
semantic-segmentation-on-mcubes-pMMSFormer (RGB-A-D)
mIoU: 52.03
semantic-segmentation-on-mcubes-pMMSFormer (RGB)
mIoU: 50.44
thermal-image-segmentation-on-pst900MMSFormer
mIoU: 87.45

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp