Automatic picture toning method based on generative adversarial network

Abstract

The invention discloses an automatic picture toning method based on a generative adversarial network. The method comprises the following steps: 1) obtaining a training group; 2) carrying out trimmingprocessing on each training group by utilizing a generator network; 3) calculating the confrontation loss of the reconstructed picture and the target picture by using a discriminator network; 4) feeding back the adversarial loss to the discriminator network to update the weight of the discriminator network; 5) calculating perception loss by using a VGG network; 6) taking the weighted sum of the perception loss and the adversarial loss as the total loss, performing back propagation to the generator model, and guiding the generator model to adjust the parameters of the generator model for performing trimming processing on the original picture; 7) repeating the operations above until training is finished; and 8), performing picture toning by means of the trained generator model. The inventionfurther discloses a picture automatic color matching system based on the generative adversarial network, automatic color matching can be achieved, the color adjusting effect is unified, the color adjusting style is stable, the final finished product pixel is high, and the calculated amount is small. The picture automatic color matching method and the system based on the generative adversarial network have the advantages that the color matching is automatic, the color adjusting effect is uniform and the color adjusting style is stable.

Description

technical field [0001] The invention belongs to the technical field of image processing, and in particular relates to the adjustment of image color, in particular to a method for automatically color-grading a picture based on a generative confrontation network. Background technique [0002] At present, the image processing of wedding photography scenes on the market is manually retouched, and the photoshop macro is used to customize the color correction module to realize the automatic process of image enhancement. The image processing efficiency is low, and the consistency of the final output effect cannot be guaranteed. . Therefore, there is a need for a software that replaces the process of manually using photoshop color grading in wedding photography, and automates color grading according to the scene of the picture and the characters. Contents of the invention [0003] In order to overcome the deficiencies of the prior art, the present invention provides a picture col...

AI Technical Summary

Problems solved by technology

[0002] At present, the image processing of wedding photography scenes on the market is manually retouched, and the photoshop macro is used to customize the color correction module to realize the automatic process of image enhancement. The image processing efficiency is low, and the consistency of the final output effect cannot be guaranteed.

Images