IC-GAN: An Improved Conditional Generative Adversarial Network for RGB-to-IR image translation with applications to forest fire monitoring

被引:20
作者
Boroujeni, Sayed Pedram Haeri [1 ]
Razi, Abolfazl [1 ]
机构
[1] Clemson Univ, Sch Comp, Clemson, SC 29632 USA
基金
美国国家科学基金会;
关键词
Deep learning; Unmanned aerial vehicle (UAV); Wildfire monitoring; Generative Adversarial Network (GAN); Image translation; Thermal images; WILDFIRE;
D O I
10.1016/j.eswa.2023.121962
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
This paper introduces a novel Deep Learning (DL) architecture for inferring temperature information from aerial true-color RGB images by transforming them into Infrared Radiation (IR) domain. This work is motivated by a few facts. First, off-the-shelf contemporary drones are typically equipped only with regular cameras. Second, IR heat-mapping cameras are costly and heavy for payload-limited drones. Third, additional communication channels and power supply would be needed when including IR cameras. Finally, IR cameras provide lower resolution and shorter distance ranges than RGB cameras. Therefore, learning-based translation of aerial IR recordings to RGB images can be extremely useful not only for new tests but also for offline processing of the currently available forest fire datasets with RGB images. We offer an Improved Conditional Generative Adversarial Network (IC-GAN), where matched IR images are used as a condition to guide the translation process by the generator. The U-Net-based generator is concatenated with a mapper module to transform the output into a stack of diverse color spaces with learnable parameters. To avoid the unnecessary penalization of pixel-level disparities and achieve structural similarity, we include clustering alignment to the loss function. The proposed framework is compared against several state-of-the-art methods, including U-Net, Efficient U-Net, GAN, and Conditional-GAN from both subjective (human perception) and objective evaluation perspectives. The results support our method's efficacy, demonstrating a significant improvement of around 6% in PSNR, 15% in UQI, 9% in SSIM, and 23% in IoU metrics.
引用
收藏
页数:16
相关论文
共 71 条
[1]  
Afghah F, 2019, IEEE CONF COMPUT, P835, DOI [10.1109/infcomw.2019.8845309, 10.1109/INFCOMW.2019.8845309]
[2]   Aerial Load Transportation With Obstacle Avoidance in Observed Environment [J].
Ashkoofaraz, Seyed Yaghoub ;
Lori, Ali Akbar Rezaei .
2022 10TH RSI INTERNATIONAL CONFERENCE ON ROBOTICS AND MECHATRONICS (ICROM), 2022, :248-253
[3]   Structural similarity index (SSIM) revisited: A data-driven approach [J].
Bakurov, Illya ;
Buzzelli, Marco ;
Schettini, Raimondo ;
Castelli, Mauro ;
Vanneschi, Leonardo .
EXPERT SYSTEMS WITH APPLICATIONS, 2022, 189
[4]   Photographic Image Synthesis with Cascaded Refinement Networks [J].
Chen, Qifeng ;
Koltun, Vladlen .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :1520-1529
[5]   Wildland Fire Detection and Monitoring Using a Drone-Collected RGB/IR Image Dataset [J].
Chen, Xiwen ;
Hopkins, Bryce ;
Wang, Hao ;
O'Neill, Leo ;
Afghah, Fatemeh ;
Razi, Abolfazl ;
Fule, Peter ;
Coen, Janice ;
Rowell, Eric ;
Watts, Adam .
IEEE ACCESS, 2022, 10 :121301-121317
[6]   Boundary IoU: Improving Object-Centric Image Segmentation Evaluation [J].
Cheng, Bowen ;
Girshick, Ross ;
Dollar, Piotr ;
Berg, Alexander C. ;
Kirillov, Alexander .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :15329-15337
[7]   StarGAN: Unified Generative Adversarial Networks for Multi-Domain Image-to-Image Translation [J].
Choi, Yunjey ;
Choi, Minje ;
Kim, Munyoung ;
Ha, Jung-Woo ;
Kim, Sunghun ;
Choo, Jaegul .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :8789-8797
[8]   Food security prediction from heterogeneous data combining machine and deep learning methods [J].
Deleglise, Hugo ;
Interdonato, Roberto ;
Begue, Agnes ;
D'Hotel, Elodie Maitre ;
Teisseire, Maguelonne ;
Roche, Mathieu .
EXPERT SYSTEMS WITH APPLICATIONS, 2022, 190
[9]  
Demir U., 2018, Patch-based image inpainting with generative adversarial networks
[10]   Ten-year wildfire mapping using satellite imagery: the case study of Western Greece [J].
Dimitriou, Eirini ;
Kyriou, Aggeliki ;
Nikolakopoulos, Konstantinos .
EARTH RESOURCES AND ENVIRONMENTAL REMOTE SENSING/GIS APPLICATIONS XIII, 2022, 12268