Fork me on GitHub

基于 ModelArts 的 StyleGAN3 生成高清图丨【华为云至简致远】

【摘要】 ModelArts简介ModelArts 是面向开发者的一站式 AI 开发平台,为机器学习与深度学习提供海量数据预处理及交互式智能标注、大规模分布式训练、自动化模型生成,及端-边-云模型按需部署能力,帮助用户快速创建和部署模型,管理全周期 AI 工作流。另外,ModelArts具有低门槛,高效率,高性能,易运维的特点。即ModelArts能够开箱即用,零基础3步即可构建AI模型;AI 开发...

ModelArts简介

ModelArts 是面向开发者的一站式 AI 开发平台,为机器学习与深度学习提供海量数据预处理及交互式智能标注、大规模分布式训练、自动化模型生成,及端-边-云模型按需部署能力,帮助用户快速创建和部署模型,管理全周期 AI 工作流。另外,ModelArts具有低门槛,高效率,高性能,易运维的特点。即ModelArts能够开箱即用,零基础3步即可构建AI模型;AI 开发全流程可视化管理,生产效率百倍提升,灵活支持多厂商多框架多功能模型统一纳管。

下面使用 ModelArts,以StyleGAN3为基础,生成高清图。

第一步,点击链接进入 ACGAN-动漫头像自动生成 的案例页面,并完成基础配置。

https://developer.huaweicloud.com/develop/aigallery/notebook/detail?id=e62562a1-29d1-4c6b-a057-b7562f44c642

点击 Run in ModelArts,进入 JupyterLab 页面。

image.png

JupyterLab 页面:

image.png

按照 基于ModelArts的动漫头像自动生成 步骤,配置当前运行环境,进行 规格切换。

第二步,安装依赖

选中以下代码,点击运行。

image.png

这段代码运行时间比较长,等待运行完成后,结果如下:

image.png

第三步,生成图片

运行以下代码:

  • 使用预训练的AFHQv2模型生成图片
# 使用预训练的AFHQv2模型生成图片
!python gen_images.py --outdir=out_0 --trunc=1 --seeds=2 \
    --network=stylegan3-r-afhqv2-512x512.pkl
# 展示效果
display(Image(filename='./out_0/seed0002.png'))  

image.png

  • 使用预训练的FFHQ模型生成图片
# 使用预训练的FFHQ模型生成图片
!python gen_images.py --outdir=out_1 --trunc=1 --seeds=2 \
    --network=stylegan3-r-ffhq-1024x1024.pkl
# 展示效果
display(Image(filename='./out_1/seed0002.png')) 
  • 使用预训练的metfaces模型生成图片
# 使用预训练的metfaces模型生成图片
!python gen_images.py --outdir=out_2 --trunc=1 --seeds=2 \
    --network=stylegan3-r-metfaces-1024x1024.pkl
# 展示效果
display(Image(filename='./out_2/seed0002.png')) 

第四步,生成插值视频

分别使用预训练的 AFHQv2模型、FFHQ模型、metfaces模型 生成视频。

  • AFHQv2模型

image.png

# 展示效果
outpath = "lerp.mp4"
mp4 = open(outpath,'rb').read()
data_url = "data:video/mp4;base64," + b64encode(mp4).decode()
HTML("""
<video width=400 controls>
      <source src="%s" type="video/mp4">
</video>
""" % data_url)
  • FFHQ模型

image.png

# 展示效果
outpath = "ffhq.mp4"
mp4 = open(outpath,'rb').read()
data_url = "data:video/mp4;base64," + b64encode(mp4).decode()
HTML("""
<video width=400 controls>
      <source src="%s" type="video/mp4">
</video>
""" % data_url)
  • metfaces模型

image.png

# 展示效果
outpath = "met.mp4"
mp4 = open(outpath,'rb').read()
data_url = "data:video/mp4;base64," + b64encode(mp4).decode()
HTML("""
<video width=400 controls>
      <source src="%s" type="video/mp4">
</video>
""" % data_url)

——END——


参考文献



本文地址:https://www.6aiq.com/article/1660583902477
本文版权归作者和AIQ共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出