Sketch-photo conversion method based on WGAN-GP and U-NET

The invention discloses a sketch-photo conversion method based on WGAN-GP and U-NET, and belongs to the field of image processing and heterogeneous image conversion. The sketch-photo conversion methodcomprises the steps of firstly obtaining a face sketch-picture database FERET, CUHK and IIIT-D for carrying out picture clipping and picture size adjustment, then carrying out data enhancement on thedata, and finally using a WGAN-GP and a U-NET for generating the pictures corresponding to sketches in a test set. According to the method, the problems of gradient explosion and gradient disappearance are solved by using the WGAN-GP, and the face photos can be generated better; the u-NET is used as a generator structure, and the details of the pictures are better restored by using a U-shaped structure; by using the patch discriminator, the number of parameters is reduced, and the generation of high-quality pictures is facilitated. According to the method, a realer photo can be generated, thedetails are complete, and