Dafnet: disentanglement-and-fusion network for arbitrary image style transfer
摘要
With the advancement of neural networks, various elaborately designed models have achieved great success in the field of style transfer. Previous studies have focused on aligning the statistical information of content and style images. Although they have addressed the disentanglement of content and style in images, existing disentanglement methods still suffer from poor separation performance and difficulty in balancing multi-scale feature fusion. To further optimize disentanglement performance and improve stylization quality, this paper designs a novel disentanglement module and fusion module. The disentanglement module disentangles the content and style components of an image. On this basis, the fusion module integrates multi-scale content information from the content image and multi-scale style information from the style image. The fused features are fed into a decoder to generate stylized images. The disentanglement capability of the disentanglement module is enhanced by introducing a disentanglement loss. Extensive experiments demonstrate the effectiveness of the proposed method. Our code used in the paper is available at https://github.com/cjhjbj/DAFNet.