A grayscale mapping method for infrared images based on generative adversarial networks

Abstract

The grayscale mapping of infrared images is an important research direction in the field of infrared image visualization. The grayscale mapping method of infrared images directly determines important visualization indicators such as detail preservation and overall perception of the original infrared images and can be considered as the foundation and guarantee for detail enhancement. Although the current mainstream grayscale mapping methods for infrared images can achieve good mapping results, there is still room for improvement in terms of preserving image details and enhancing image contrast. In this paper, we propose a grayscale mapping method for infrared images based on generative adversarial networks. Firstly, our discriminator adopts a unique global-local structure, which allows the network to consider both global and local losses when calculating the loss, effectively improving the image quality in local regions of the mapped image. Secondly, we introduce perceptual loss in the loss function, which ensures that the generated image and the target image have consistent features as much as possible. We conducted subjective and objective evaluations on the mapping results of our method and eight mainstream methods. The evaluation results show that our method is superior in terms of preserving image details and enhancing image contrast. Further comparison with a parameter-free tone mapping operator using generative adversarial network (TMO-Net) indicates that our method avoids problems such as target edge blur and artifacts in the mapped images, resulting in higher visual quality of the mapped images.