An Unsupervised Colorization Method of Thermal Infrared Image Based on Edge Consistency

Jiaming cai,Xin Tang,Yao Hu,Shaohui Zhang
DOI: https://doi.org/10.1117/12.3004945
2023-01-01
Abstract:Due to the lack of pixel level structural matching data, thermal infrared grayscale images are more difficult to color than visible and near-infrared grayscale images. Therefore, this paper proposes a unsupervised learning method based on CycleGAN. On the basis of CycleGAN, a pre trained edge monitor is introduced to calculate the edge feature map before and after image transformation, and the edge similarity loss function is calculated as the basis for optimizing the neural network parameters. The experimental results show that the proposed method effectively reduces the loss of effective edge information during the coloring process and suppresses the generation of abnormal edge information during the coloring process.
What problem does this paper attempt to address?