Application of Generative Adversarial Network in Image Color Correction

Abstract

The popularity of electronic products has increased with the development of technology. Electronic devices allow people to obtain information through the transmission of images. However, color distortion can occur during the transmission process, which may hinder the usefulness of the images. To this end, a deep residual network and a deep convolutional network were used to define the generator and discriminator. Then, self-attention-enhanced convolution was applied to the generator network to construct an image resolution correction model based on coupled generative adversarial networks. On this basis, a generative network model integrating multi-scale features and contextual attention mechanism was constructed to achieve image restoration. Finally, performance and image restoration application tests were conducted on the constructed model. The test showed that when the coupled generative adversarial network was tested on the Set5 dataset, the image peak signal-to-noise ratio and image structure similarity values were 31.2575 and 0.8173. On the Set14 dataset, they were 30.8521 and 0.8079, respectively. The multi-scale feature-fusion algorithm was tested on the BSDS100 dataset with an image peak signal-to-noise ratio of 30.2541 and an image structure similarity value of 0.8352. Based on the data presented, it can be concluded that the image correction model constructed in this study has a strong image restoration ability. The reconstructed image has the highest similarity with the real high-resolution image and a low distortion rate. It can achieve the task of repairing problems such as color distortion during image transmission. In addition, this study can provide technical support for similar information correction and restoration work.