一种基于权重解调的图像风格迁移方法及装置
Abstract:
本发明提供一种基于权重解调的图像风格迁移方法及装置。方法包括:S1、将所述主体内容图像和风格图像分别输入编码网络进行特征提取处理;S2、将所述主体内容特征和风格特征输入调制解调模块进行处理,其中所述调制解调模块包括若干结构相同的调制解调子模块,任意所述调制解调子模块包括卷积层,所述卷积层的输入数据为调制解调子模块接收的主体内容特征,所述卷积层的输出数据作为下一个调制解调子模块中卷积层的输入数据;S3、将所述融合特征传入解码网络进行特征还原处理,从而获得同时具有主体内容以及迁移风格的风格迁移图像。本发明通过应用调制解调的方法,实现了在轻量化网络以及小样本训练条件下,生成更高质量的风格迁移图像。
Public/Granted literature
Patent Agency Ranking
0/0