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}
}