先上 demo:
本项目是我拿来快速部署 PyTorch 模型的一个模板项目,上面的 GIF 中展示的模型为在 AFHQ 数据集上预训练好的 StarGANv2 模型,如你所见可以拿来转换动物图片。
本来想在 Heroku 上部署一个让大家可以立刻试玩的,但是由于 PyTorch 太大(~700M )超出了 Heroku 的白嫖计划的容量闲置,于是只能上 GIF 展示了。
仓库地址: https://github.com/songquanpeng/pytorch-deployment
与之配套的模板项目还有: https://github.com/songquanpeng/pytorch-template (这个是拿来快速搭建 PyTorch 项目的)。
1
JustSong OP 额,为啥这么靠下,难道我被降权了😬
|