Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

UniRGB-IR: A Unified Framework for Visible-Infrared Semantic Tasks via Adapter Tuning

About

Semantic analysis on visible (RGB) and infrared (IR) images has gained significant attention due to their enhanced accuracy and robustness under challenging conditions including low-illumination and adverse weather. However, due to the lack of pre-trained foundation models on the large-scale infrared image datasets, existing methods prefer to design task-specific frameworks and directly fine-tune them with pre-trained foundation models on their RGB-IR semantic relevance datasets, which results in poor scalability and limited generalization. To address these limitations, we propose UniRGB-IR, a scalable and efficient framework for RGB-IR semantic tasks that introduces a novel adapter mechanism to effectively incorporate rich multi-modal features into pre-trained RGB-based foundation models. Our framework comprises three key components: a vision transformer (ViT) foundation model, a Multi-modal Feature Pool (MFP) module, and a Supplementary Feature Injector (SFI) module. The MFP and SFI modules cooperate with each other as an adpater to effectively complement the ViT features with the contextual multi-scale features. During training process, we freeze the entire foundation model to inherit prior knowledge and only optimize the MFP and SFI modules. Furthermore, to verify the effectiveness of our framework, we utilize the ViT-Base as the pre-trained foundation model to perform extensive experiments. Experimental results on various RGB-IR semantic tasks demonstrate that our method can achieve state-of-the-art performance.

Maoxun Yuan, Bo Cui, Tianyi Zhao, Jiayi Wang, Shan Fu, Xue Yang, Xingxing Wei• 2024

Related benchmarks

TaskDatasetResultRank
Object DetectionFLIR (test)
mAP500.814
83
Object DetectionLLVIP
mAP5096.1
58
Object DetectionFLIR
mAP44.1
40
Multispectral Pedestrian DetectionKAIST (test)--
39
Object DetectionLLVIP (test)
mAP5096.1
38
Pedestrian DetectionKAIST multispectral pedestrian dataset (All-dataset)
Detection Rate (Near)0.00e+0
14
Salient Object DetectionVT821 (test)
S-Measure0.881
13
Salient Object DetectionVT1000 (test)
S-Measure93.9
13
Salient Object DetectionVT5000 (test)
S-Measure0.906
13
Showing 9 of 9 rows

Other info

Code

Follow for update