arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.03748cs.CV

基于边缘引导对抗条件扩散模型的热模态数据生成方案

Data Generation Scheme for Thermal Modality with Edge-Guided Adversarial Conditional Diffusion Model

Harbin Institute of Technology, Shenzhen(哈尔滨工业大学(深圳));University of Macau(澳门大学)

Guoqing Zhu, Honghu Pan, Qiang Wang, Chao Tian, Chao Yang, Zhenyu He

AI总结:

本文提出边缘引导条件扩散模型ECDM,利用可见光边缘信息生成像素级对齐的伪热图像,并通过两阶段模态对抗训练过滤可见光特有边缘,在LLVIP上取得优于现有方法的图像生成质量。

摘要:

在具有挑战性的低光照和恶劣天气条件下,热视觉算法,尤其是目标检测,已展现出显著潜力,这与可见光视觉算法频繁遭遇的困境形成鲜明对比。然而,由深度学习模型驱动的热视觉算法的有效性仍然受到可用训练数据样本稀缺的制约。为此,本文提出了一种名为边缘引导条件扩散模型的新方法。该框架旨在利用从可见光图像中提取的边缘信息,生成像素级精确对齐的伪热图像。通过利用边缘作为来自可见光域的上下文线索,扩散模型实现了对生成图像中目标轮廓的精细控制。为了减轻那些不应出现在热域中的可见光特有边缘信息的影响,提出了一种两阶段模态对抗训练策略,通过区分可见光和热模态,将这些信息从生成图像中过滤掉。在LLVIP上的大量实验表明,ECDM在图像生成质量方面优于现有的最先进方法。

英文摘要:

In challenging low light and adverse weather conditions,thermal vision algorithms,especially object detection,have exhibited remarkable potential,contrasting with the frequent struggles encountered by visible vision algorithms. Nevertheless,the efficacy of thermal vision algorithms driven by deep learning models remains constrained by the paucity of available training data samples. To this end,this paper introduces a novel approach termed the edge guided conditional diffusion model. This framework aims to produce meticulously aligned pseudo thermal images at the pixel level,leveraging edge information extracted from visible images. By utilizing edges as contextual cues from the visible domain,the diffusion model achieves meticulous control over the delineation of objects within the generated images. To alleviate the impacts of those visible-specific edge information that should not appear in the thermal domain,a two-stage modality adversarial training strategy is proposed to filter them out from the generated images by differentiating the visible and thermal modality. Extensive experiments on LLVIP demonstrate ECDM s superiority over existing state-of-the-art approaches in terms of image generation quality.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑