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

Text-DiFuse: An Interactive Multi-Modal Image Fusion Framework based on Text-modulated Diffusion Model

About

Existing multi-modal image fusion methods fail to address the compound degradations presented in source images, resulting in fusion images plagued by noise, color bias, improper exposure, \textit{etc}. Additionally, these methods often overlook the specificity of foreground objects, weakening the salience of the objects of interest within the fused images. To address these challenges, this study proposes a novel interactive multi-modal image fusion framework based on the text-modulated diffusion model, called Text-DiFuse. First, this framework integrates feature-level information integration into the diffusion process, allowing adaptive degradation removal and multi-modal information fusion. This is the first attempt to deeply and explicitly embed information fusion within the diffusion process, effectively addressing compound degradation in image fusion. Second, by embedding the combination of the text and zero-shot location model into the diffusion fusion process, a text-controlled fusion re-modulation strategy is developed. This enables user-customized text control to improve fusion performance and highlight foreground objects in the fused images. Extensive experiments on diverse public datasets show that our Text-DiFuse achieves state-of-the-art fusion performance across various scenarios with complex degradation. Moreover, the semantic segmentation experiment validates the significant enhancement in semantic performance achieved by our text-controlled fusion re-modulation strategy. The code is publicly available at https://github.com/Leiii-Cao/Text-DiFuse.

Hao Zhang, Lei Cao, Jiayi Ma• 2024

Related benchmarks

TaskDatasetResultRank
Semantic segmentationMFNet (test)
mIoU59.93
134
Semantic segmentationFMB (test)
mIoU56.48
59
Infrared-Visible Image FusionRoadScene (test)
Average Gradient (AG)2.96
40
Object DetectionLLVIP (test)
mAP5093.9
38
Object DetectionMSRS (test)
mAP@0.589.7
34
Infrared-Visible Image FusionLLVIP (test)
EN7.08
23
Image FusionMSRS (test)
Entropy7.08
10
Image FusionHarvard Medicine Dataset (test)
Entropy (EN)6.44
10
Visible-Infrared Image FusionMSRS
EI10.36
10
Visual Representation GeneralizationFMB (test)
EN6.93
10
Showing 10 of 24 rows

Other info

Code

Follow for update