Thermal Image Calibration and Correction using Unpaired Cycle-Consistent Adversarial Networks
Hossein Rajoli, Pouya Afshin, Fatemeh Afghah
TL;DR
This study tackles the scarcity and heterogeneity of aerial wildfire datasets by proposing a CycleGAN-based calibration framework that translates low-quality IR images to a high-quality, temperature-referenced domain. The model employs non-symmetric generators with multi-level feature fusion and RGB attribute conditioning to preserve content while enhancing fidelity, guided by a composite loss that includes adversarial, cycle, identity, perceptual, and SSIM components. Evaluation on Flame II and Flame III datasets shows improved structural similarity and perceptual fidelity over baselines, indicating better alignment with high-tech camera outputs. The approach enables harmonization of disparate wildfire imagery, facilitating scalable DL-based wildfire analysis across differing sensor platforms and improving data utility for detection, characterization, and management tasks.
Abstract
Unmanned aerial vehicles (UAVs) offer a flexible and cost-effective solution for wildfire monitoring. However, their widespread deployment during wildfires has been hindered by a lack of operational guidelines and concerns about potential interference with aircraft systems. Consequently, the progress in developing deep-learning models for wildfire detection and characterization using aerial images is constrained by the limited availability, size, and quality of existing datasets. This paper introduces a solution aimed at enhancing the quality of current aerial wildfire datasets to align with advancements in camera technology. The proposed approach offers a solution to create a comprehensive, standardized large-scale image dataset. This paper presents a pipeline based on CycleGAN to enhance wildfire datasets and a novel fusion method that integrates paired RGB images as attribute conditioning in the generators of both directions, improving the accuracy of the generated images.
