StawGAN: Structural-Aware Generative Adversarial Networks for Infrared Image Translation

Luigi Sigillo,Eleonora Grassucci,Danilo Comminiello
2023-05-18
Abstract:This paper addresses the problem of translating night-time thermal infrared images, which are the most adopted image modalities to analyze night-time scenes, to daytime color images (NTIT2DC), which provide better perceptions of objects. We introduce a novel model that focuses on enhancing the quality of the target generation without merely colorizing it. The proposed structural aware (StawGAN) enables the translation of better-shaped and high-definition objects in the target domain. We test our model on aerial images of the DroneVeichle dataset containing RGB-IR paired images. The proposed approach produces a more accurate translation with respect to other state-of-the-art image translation models. The source code is available at <a class="link-external link-https" href="https://github.com/LuigiSigillo/StawGAN" rel="external noopener nofollow">this https URL</a>
Computer Vision and Pattern Recognition,Machine Learning,Image and Video Processing
What problem does this paper attempt to address?
The paper aims to address the problem of converting nighttime thermal infrared images to daytime color images, specifically the translation task from nighttime thermal infrared images to daytime color images (NTIT2DC). Specifically, the paper proposes a new model—Structure-Aware Generative Adversarial Network (StawGAN)—to improve the quality of generated targets rather than simply coloring the images. StawGAN generates better-shaped and more detailed target objects by sharing connections to transfer information between images and segmentation targets. The researchers tested the model on the DroneVehicle dataset, which contains RGB-IR paired images, and demonstrated that it outperforms existing image translation models on various evaluation metrics. Additionally, the paper showcases the potential application value of StawGAN in downstream tasks such as object detection and classification.