中文

StyleAutoEncoder:用于操控 StyleGAN 前训练图像属性的自编码器

计算机视觉与模式识别 2024-12-31 v1 人工智能

摘要

深度条件生成模型是创建高质量图像和编辑其属性的优秀工具。然而,零散地训练现代生成模型需要巨大的计算资源。在本文中,我们引入 StyleAutoEncoder (StyleAE),一种轻量级自编码器模块,作为预训练生成模型的插件,允许对图像的所需属性进行操控。该方法为在计算资源有限的情况下训练深度生成模型提供了具成本效益的解决方案,为广泛应用提供了前景。我们通过将 StyleAE 与目前顶级生成模型之一 StyleGAN 结合来评估 StyleAE。我们的实验表明,StyleAE 在操控图像属性方面至少与基于可逆正规化流的领先算法一样有效,但更为简单、更快,且在神经网络设计上提供了更多自由度。

关键词

引用

@article{arxiv.2412.20164,
  title  = {StyleAutoEncoder for manipulating image attributes using pre-trained StyleGAN},
  author = {Andrzej Bedychaj and Jacek Tabor and Marek Śmieja},
  journal= {arXiv preprint arXiv:2412.20164},
  year   = {2024}
}