Aiming at the problem that the existing methods can only extract the low-order statistics of image features when implementing the task of style transfer, this study considered modeling the style transfer process as a feature distribution matching process, proposed a discriminator network based on Wasserstein distance, and defined a style loss function. Wasserstein discriminator can better fit the Wasserstein distance between feature distributions, and the defined style loss can better distinguish the difference between higher-order statistical information of image features. At the same time, in order to achieve the effect of real-time generation, a style transfer conversion module based on an encoder-decoder structure and an attention mechanism was introduced as a generation network. This generation network can effectively integrate the original image features and generate them. Specifically, the style loss was computed by adding a Wasserstein discriminator after the convolutional layer of the computational loss module. The training of the generative network was then supervised by the style loss together with the content loss calculated as the mean square error in the traditional methods, and after the network training, images can be input for the style transfer test. Finally, the network was trained on the benchmark MSCOCO and WikiArt datasets and the results were tested. The qualitative and quantitative experimental results showed that the proposed method can achieve real-time style transfer and generate high-quality stylization results compared with existing methods.
参考文献
相似文献
引证文献
引用本文
米志鹏.基于风格特征分布匹配的图像风格迁移[J].智能科学与工程学报,2023,40(2):51-56 MI Zhipeng. Image Style Transfer Based on the Distribution Matching of the Style Features[J]. Journal of Chongqing Technology and Business University(Natural Science Edition),2023,40(2):51-56