HyperAIHyperAI

Command Palette

Search for a command to run...

对抗性引导的人像抠图

Sergej Chicherin Karen Efremyan

摘要

我们提出了一种基于有限数据源生成 alpha 蒙版的方法。首先,我们在人像数据集上对一种新型基于 Transformer 的模型(StyleMatte)进行预训练。随后,利用该模型生成图像-掩码配对数据,用于训练基于 StyleGAN3 架构的网络(StyleMatteGAN)。该网络采用无监督方式训练,并生成此前未见过的图像-掩码训练样本,反馈回 StyleMatte 模型以实现迭代优化。实验表明,该循环过程显著提升了蒙版提取网络的性能,在人像数据集上取得了顶尖结果,并在动物图像数据集上达到了当前最先进的评估指标。此外,StyleMatteGAN 能够生成高分辨率、具备隐私保护特性的带 alpha 蒙版人像图像,适用于多种图像合成任务。相关代码已开源,地址为:https://github.com/chroneus/stylematte


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供