中文

MobileStyleGAN:一种用于高保真图像生成的轻量级卷积神经网络

计算机视觉与模式识别 2021-09-13 v2 图像与视频处理

摘要

近年来,生成对抗网络(GANs)在生成图像建模中的使用变得非常流行。虽然基于风格的 GAN 架构在高保真图像生成中取得了最优结果,但在计算上它们高度复杂。在我们的工作中,我们专注于基于风格的生成模型的性能优化。我们分析了 StyleGAN2 中计算最繁重的部分,并提出对生成器网络进行改动,以使基于风格的生成网络能够部署在边缘设备中。我们引入了 MobileStyleGAN 架构,其参数量比 StyleGAN2 少 3.5 倍,计算复杂度低 9.5 倍,同时提供可比的质量。

关键词

引用

@article{arxiv.2104.04767,
  title  = {MobileStyleGAN: A Lightweight Convolutional Neural Network for High-Fidelity Image Synthesis},
  author = {Sergei Belousov},
  journal= {arXiv preprint arXiv:2104.04767},
  year   = {2021}
}