之前推荐过一个基于 TensorFlow 的 GAN 框架–谷歌开源的 GAN 库–TFGAN。
而最近也有一个新的 GAN 框架工具,并且是基于 Pytorch 实现的,项目地址如下:
https://github.com/torchgan/torchgan
对于习惯使用 Pytorch 框架的同学,现在可以采用这个开源项目快速搭建一个 GAN 网络模型了!
目前该开源项目有 400+ 星,它给出了安装的教程、API 文档以及使用教程,文档的地址如下:
https://torchgan.readthedocs.io/en/latest/
安装
对于 TorchGAN 的安装,官网给出 3 种方法,但实际上目前仅支持两种安装方式,分别是pip
方式安装以及源码安装,采用conda
安装的方法目前还不支持。