arXiv:2504.02801cs.CV2025-04被引 8

用大模型引导扩散模型,实现可见光到热成像的高质量转换

F-ViTA: Foundation Model Guided Visible to Thermal Translation

论文配图:F-ViTA: Foundation Model Guided Visible to Thermal Translation
图 1 · 摘自论文原文
  • 利用大模型零样本提示生成掩码和标签,指导扩散过程
  • 在五个公开数据集上超越现有最佳方法,跨波段生成效果好
  • 适合需要低成本热成像生成的研究者与工程师

热成像对场景理解至关重要,尤其在低光照和夜间条件下。然而,由于红外图像采集需专用设备,大规模热成像数据集的构建成本高、耗时长。为此,研究者探索了可见光到热成像的图像翻译任务。现有方法多基于生成对抗网络(GAN)或扩散模型(DM),将该任务视为风格迁移,试图从有限训练数据中同时学习模态分布偏移和物理规律。本文提出F-ViTA,一种新方法,利用嵌入于基础模型中的通用世界知识引导扩散过程以提升翻译质量。具体而言,我们使用SAM和Grounded DINO等基础模型生成的零样本掩码和标签,对InstructPix2Pix扩散模型进行条件控制,使其学习场景物体与其红外热特征之间的有意义关联。在五个公开数据集上的大量实验表明,F-ViTA优于当前最先进方法。此外,模型在分布外(OOD)场景下具有良好的泛化能力,可从同一可见光图像生成长波红外(LWIR)、中波红外(MWIR)和近红外(NIR)图像。代码已开源。

原文摘要 · Abstract (English)

Thermal imaging is crucial for scene understanding, particularly in low-light and nighttime conditions. However, collecting large thermal datasets is costly and labor-intensive due to the specialized equipment required for infrared image capture. To address this challenge, researchers have explored visible-to-thermal image translation. Most existing methods rely on Generative Adversarial Networks (GANs) or Diffusion Models (DMs), treating the task as a style transfer problem. As a result, these approaches attempt to learn both the modality distribution shift and underlying physical principles from limited training data. In this paper, we propose F-ViTA, a novel approach that leverages the general world knowledge embedded in foundation models to guide the diffusion process for improved translation. Specifically, we condition an InstructPix2Pix Diffusion Model with zero-shot masks and labels from foundation models such as SAM and Grounded DINO. This allows the model to learn meaningful correlations between scene objects and their thermal signatures in infrared imagery. Extensive experiments on five public datasets demonstrate that F-ViTA outperforms state-of-the-art (SOTA) methods. Furthermore, our model generalizes well to out-of-distribution (OOD) scenarios and can generate Long-Wave Infrared (LWIR), Mid-Wave Infrared (MWIR), and Near-Infrared (NIR) translations from the same visible image. Code: https://github.com/JayParanjape/F-ViTA/tree/master.

图像翻译扩散模型热成像基础模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。