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

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhu, Guoqing, Pan, Honghu, Wang, Qiang, Tian, Chao, Yang, Chao, He, Zhenyu
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913460538834944
author Zhu, Guoqing
Pan, Honghu
Wang, Qiang
Tian, Chao
Yang, Chao
He, Zhenyu
author_facet Zhu, Guoqing
Pan, Honghu
Wang, Qiang
Tian, Chao
Yang, Chao
He, Zhenyu
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2408_03748
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Data Generation Scheme for Thermal Modality with Edge-Guided Adversarial Conditional Diffusion Model
Zhu, Guoqing
Pan, Honghu
Wang, Qiang
Tian, Chao
Yang, Chao
He, Zhenyu
Computer Vision and Pattern Recognition
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.
title Data Generation Scheme for Thermal Modality with Edge-Guided Adversarial Conditional Diffusion Model
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.03748