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

[JustSong]

前言

先上 demo：

简单介绍

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

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

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

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

[JustSong]

额，为啥这么靠下，难道我被降权了😬

[Aether]

@JustSong 方便求一个零基础小白部署教程吗？

[Aether]

@JustSong 所以这是一个部署好的 flask 网站，只需要安装组件就可以跑起来了，是吗？

[JustSong]

@Aether 是的，不过模型权重是从 Github 上下载的，如果服务器访问不了 Github，则需要手动下载并放到指定目录