arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.11010cs.CV

MM2Latent:多模态辅助下基于GAN的文本到人脸图像生成与编辑

MM2Latent: Text-to-facial image generation and editing in GANs with multimodal assistance

Queen Mary University of London(伦敦玛丽女王大学);Samsung AI Center(三星人工智能中心)

Debin Meng, Christos Tzelepis, Ioannis Patras, Georgios Tzimiropoulos

AI总结:

针对单模态人脸生成可控性不足、现有多模态方法存在超参数多、推理需人工干预等问题,本文提出MM2Latent框架,通过映射网络将多模态输入映射至StyleGAN的w潜在空间,实现了高效可控的多模态人脸生成与编辑,性能优于现有GAN及扩散类方法。

摘要:

人脸生成是图像生成领域的热点方向,例如掩码到人脸生成、文本到人脸生成等。但这类单模态生成方法在图像生成过程中缺乏可控性,而挖掘不同模态的优势与互补性能够提升可控性——比如可利用文本在控制多样属性上的优势,以及掩码在控制空间位置上的优势。当前多模态生成领域的最优方法存在诸多局限:依赖大量超参数、推理阶段需人工操作、训练与推理阶段计算需求过高,或是无法编辑真实图像。本文提出了一种实用的多模态图像生成与编辑框架MM2Latent。我们以StyleGAN2作为图像生成器,采用FaRL进行文本编码,并针对掩码、草图、3DMM等空间模态训练了一个自编码器。我们提出了一种训练映射网络的策略,将多模态输入映射到StyleGAN的w潜在空间中。该框架具备三点特性:1)消除了推理阶段的超参数与人工操作;2)保证了较快的推理速度;3)支持真实图像编辑。大量实验表明,本文方法在多模态图像生成任务上表现优异,性能超越了近期基于GAN和扩散模型的方法;同时其在多模态图像编辑任务中同样有效,且推理速度快于基于GAN和扩散模型的方法。代码已公开在:https://github.com/Open-Debin/MM2Latent

英文摘要:

Generating human portraits is a hot topic in the image generation area, e.g. mask-to-face generation and text-to-face generation. However, these unimodal generation methods lack controllability in image generation. Controllability can be enhanced by exploring the advantages and complementarities of various modalities. For instance, we can utilize the advantages of text in controlling diverse attributes and masks in controlling spatial locations. Current state-of-the-art methods in multimodal generation face limitations due to their reliance on extensive hyperparameters, manual operations during the inference stage, substantial computational demands during training and inference, or inability to edit real images. In this paper, we propose a practical framework - MM2Latent - for multimodal image generation and editing. We use StyleGAN2 as our image generator, FaRL for text encoding, and train an autoencoders for spatial modalities like mask, sketch and 3DMM. We propose a strategy that involves training a mapping network to map the multimodal input into the w latent space of StyleGAN. The proposed framework 1) eliminates hyperparameters and manual operations in the inference stage, 2) ensures fast inference speeds, and 3) enables the editing of real images. Extensive experiments demonstrate that our method exhibits superior performance in multimodal image generation, surpassing recent GAN- and diffusion-based methods. Also, it proves effective in multimodal image editing and is faster than GAN- and diffusion-based methods. We make the code publicly available at: https://github.com/Open-Debin/MM2Latent

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑