基于目标特征调色板与注意力着色的风格迁移
计算机视觉与模式识别
2021-11-09 v1 图像与视频处理
摘要
风格迁移吸引了大量关注，因为它能将给定图像转变为具有华丽艺术风格且保留图像结构的图像。然而，传统方法在风格迁移过程中容易丢失图像细节并产生不悦目的伪影。在本文中，为解决这些问题，提出了一种基于目标特征调色板的新型艺术风格化方法，其可准确迁移关键特征。具体而言，我们的方法包含两个模块，即特征调色板组合（FPC）和注意力着色（AC）模块。FPC 模块基于 K-means 聚类捕获代表性特征并生成特征目标调色板。随后的 AC 模块计算内容图与风格图之间的注意力图，并基于注意力图和目标调色板迁移颜色与图案。这些模块使所提出的风格化能够聚焦于关键特征并生成合理的迁移图像。因此，所提方法的贡献在于提出一种基于深度学习的新颖风格迁移方法并给出目标特征调色板和注意力着色模块，以及通过详尽的消融研究对所提方法提供深入分析与见解。定性与定量结果表明，我们的风格化图像展现出最先进的性能，在保留内容图像核心结构与细节方面具有优势。
引用
@article{arxiv.2111.04028,
title = {Style Transfer with Target Feature Palette and Attention Coloring},
author = {Suhyeon Ha and Guisik Kim and Junseok Kwon},
journal= {arXiv preprint arXiv:2111.04028},
year = {2021}
}