Image Recognition Based on Super-Resolution Wasserstein Generative Adversarial Nets with Gradient Penalty

Abstract

Generative adversarial networks (GAN) are currently a hotly debated research topic in the field of machine vision; however, they possess various shortcomings that cannot be overlooked, such as unstable generated samples, collapsed modes, and slow convergence. The present paper combines the advantages of the super-resolution GAN model (SRGAN) and Wasserstein GAN with gradient penalty (WGAN-GP) and proposes a hybrid model, known as super-resolution Wasserstein generative adversarial network with gradient penalty (SRWGAN-GP). The generator of SRWGAN-GP utilizes the residuals block and Wasserstein distance to increase the resolution of the generated samples and solve the shortcomings of the unstable GAN training. Besides, the discriminator of SRWGAN-GP consists of a local discriminator and a global discriminator to enhance image recognition capacity. Finally, the discriminator is extracted for image recognition. The results of comparative experiments indicate that this method can effectively enhance the accuracy of image recognition.