摘要
本发明属于计算机视觉与深度学习技术领域,具体涉及一种基于亲和力增强注意力网络的图像风格迁移算法,包括下列步骤:首先,通过预训练的深度编码器提取内容图像与风格图像的多层次特征;接着,利用内容亲和力增强注意力模块和风格亲和力增强注意力模块分别对内容特征和风格特征进行局部细节强化,通过注意力机制增强纹理结构与色彩分布的表示;随后,通过混合注意力模块动态调整风格特征的空间分布,使其与内容特征语义对齐,完成自适应特征融合;最后,通过对称设计的解码器将融合特征重构为风格化图像。本发明可以在最大限度保留内容特征的基础上完成风格特征的迁移。