基于PhotoWCT2和多重注意力机制的印花图案配色模型

来源：计算机测量与控制杂志2025年第10期北京时间：

作者：汤逸烽,蒋俊豪,张津,朱威

单位：浙江工业大学 信息工程学院

摘要：针对现有风格迁移算法无法满足印花图案配色工作对纹理保留和颜色风格迁移效果的要求，设计了一种基于PhotoWCT2和多重注意力机制的印花图案配色模型；该模型首先对PhotoWCT2网络添加了多尺度通道-局部注意力模块，增强了模型对印花图案复杂纹理特征的提取能力；其次引入频率注意力模块，加强了频域中的颜色信息，提升模型对颜色风格的迁移能力；最后采用基于对比学习的内容损失函数，增强模型对特征空间的建模能力，提升模型对印花图案复杂纹理特征的重建能力。实验结果表明，与原始网络模型相比，所设计模型在SSIM、PSNR和Gram Loss这三个客观指标上均表现更优。此外，与其它网络模型相比，所设计模型的印花图案配色结果也具有更好的纹理保留和颜色风格迁移效果。

关键词：风格迁移;深度学习;注意力机制;对比学习;印花图案配色

基金资助：杭州市重大科技创新项目(2022AIZD0077);国家自然科学基金联合基金重点项目(U24A20270)