中文

HSI:一种用于任意风格迁移的整体风格注入器

计算机视觉与模式识别 2025-02-10 v1 机器学习 图像与视频处理

摘要

基于注意力的任意风格迁移方法因其合成风格细节的出色能力而近期备受关注。然而,注意力机制中的逐点匹配可能过度关注局部模式,从而忽略了风格图像显著的全局特征。此外,在处理大尺寸图像时,注意力机制的二次复杂度会带来高昂的计算负担。为了缓解上述问题,我们提出了整体风格注入器(HSI),这是一种新颖的注意力风格转换模块,用于传递目标风格的艺术表现力。具体而言,HSI 仅基于更符合风格迁移特性的全局风格表示进行风格化,以避免在风格化图像中生成局部不协调的模式。此外,我们在 HSI 内部提出了一种双重关系学习机制,通过利用内容和风格中的语义相似性来动态渲染图像,确保风格化图像保留原始内容并提高风格保真度。值得注意的是,所提出的 HSI 实现了线性计算复杂度,因为它通过逐元素乘法而非矩阵乘法建立特征映射。定性和定量结果表明,我们的方法在有效性和效率方面均优于最先进的方法。

关键词

引用

@article{arxiv.2502.04369,
  title  = {HSI: A Holistic Style Injector for Arbitrary Style Transfer},
  author = {Shuhao Zhang and Hui Kang and Yang Liu and Fang Mei and Hongjuan Li},
  journal= {arXiv preprint arXiv:2502.04369},
  year   = {2025}
}