V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
爱意满满的作品展示区。
JustSong
V2EX  ›  分享创造

分享一个可以随意转换动物图片的项目

  •  
  •   JustSong ·
    songquanpeng · 2021-08-21 20:08:06 +08:00 · 1026 次点击
    这是一个创建于 1198 天前的主题,其中的信息可能已经有所发展或是发生改变。

    前言

    先上 demo: Demo

    简单介绍

    本项目是我拿来快速部署 PyTorch 模型的一个模板项目,上面的 GIF 中展示的模型为在 AFHQ 数据集上预训练好的 StarGANv2 模型,如你所见可以拿来转换动物图片。

    本来想在 Heroku 上部署一个让大家可以立刻试玩的,但是由于 PyTorch 太大(~700M )超出了 Heroku 的白嫖计划的容量闲置,于是只能上 GIF 展示了。

    仓库地址: https://github.com/songquanpeng/pytorch-deployment

    与之配套的模板项目还有: https://github.com/songquanpeng/pytorch-template (这个是拿来快速搭建 PyTorch 项目的)。

    JustSong
        1
    JustSong  
    OP
       2021-08-21 20:11:33 +08:00 via Android
    额,为啥这么靠下,难道我被降权了😬
    Aether
        2
    Aether  
       2021-08-22 16:23:07 +08:00
    @JustSong 方便求一个零基础小白部署教程吗?
    Aether
        3
    Aether  
       2021-08-22 16:30:19 +08:00
    @JustSong 所以这是一个部署好的 flask 网站,只需要安装组件就可以跑起来了,是吗?
    JustSong
        4
    JustSong  
    OP
       2021-08-22 17:54:25 +08:00 via Android
    @Aether 是的,不过模型权重是从 Github 上下载的,如果服务器访问不了 Github,则需要手动下载并放到指定目录
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   916 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 25ms · UTC 20:34 · PVG 04:34 · LAX 12:34 · JFK 15:34
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.