计算机工程

作者：陈佳, 董学良, 梁金星, 何儒汉

单位：1. 武汉纺织大学 数学与计算机学院, 武汉 430000;2. 湖北省服装信息化工程技术研究中心, 武汉 430000

关键词：图像风格迁移,边界伪影,注意力机制,循环生成对抗网络,损失函数

基金：湖北省自然科学基金计划一般面上项目（2020CFB801）；湖北省高等学校优秀中青年科技创新团队计划（T201807）。

针对复杂背景下服装图像局部区域风格迁移难以控制及迁移后容易产生边界伪影的问题，提出一种基于注意力机制的CycleGAN服装局部风格迁移方法。通过VGG16网络分别提取服装图像的内容特征与风格特征，将其输入基于注意力机制的CycleGAN生成器中，应用注意力机制在复杂背景下的各个服装区域分配概率分布信息，获得注意力分布更多的区域及相关度更高的区域，并采用改进的损失函数校正边界伪影，对该区域进行风格迁移得到所需的风格迁移服装图像。实验结果表明，与CNN、FCN、BeautyGAN图像局部风格迁移方法相比，该方法不仅可以突出服装图像局部风格迁移效果，而且增强了图像细节，有利于提高输出图像的真实性和艺术性。

来源：2021年第11期

《计算机工程》期刊编辑部

查看计算机工程杂志2021年第11期