之前推薦過一個(gè)基于 TensorFlow 的 GAN 框架–谷歌開源的 GAN 庫–TFGAN。

而最近也有一個(gè)新的 GAN 框架工具，并且是基于 Pytorch 實(shí)現(xiàn)的，項(xiàng)目地址如下：

https://github.com/torchgan/torchgan

對于習(xí)慣使用 Pytorch 框架的同學(xué)，現(xiàn)在可以采用這個(gè)開源項(xiàng)目快速搭建一個(gè) GAN 網(wǎng)絡(luò)模型了！

目前該開源項(xiàng)目有 400+ 星，它給出了安裝的教程、API 文檔以及使用教程，文檔的地址如下：

https://torchgan.readthedocs.io/en/latest/

安裝

對于 TorchGAN 的安裝，官網(wǎng)給出 3 種方法，但實(shí)際上目前僅支持兩種安裝方式，分別是pip方式安裝以及源碼安裝，采用conda安裝的方法目前還不支持。

Pip 安裝方法

安裝最新的發(fā)布版本的命令如下：

$ pip3 install torchgan

而如果是最新版本：

$ pip3 install git+https://github.com/torchgan/torchgan.git

Conda 安裝

這是目前版本還不支持的安裝方式，將會(huì)在v0.1版本實(shí)現(xiàn)這種安裝方法。

源碼方式安裝

按照下列命令的順序執(zhí)行來進(jìn)行從源碼安裝

$ git clone https://github.com/torchgan/torchgan $ cd torchgan $ python setup.py install

依賴庫

必須按照的依賴庫：

• Numpy
• Pytorch 0.4.1
• Torchvision

可選

• TensorboardX：主要是為了采用Tensorboard來觀察和記錄實(shí)驗(yàn)結(jié)果。安裝通過命令pip install tensorboardX
• Visdom：為了采用Xisdom進(jìn)行記錄。安裝通過命令pip install visdom

API 文檔

API 的文檔目錄如下：

從目錄主要分為以下幾個(gè)大類：

• torchgan.layers：包含當(dāng)前常用的用于構(gòu)建 GAN 結(jié)構(gòu)的一些網(wǎng)絡(luò)層，包括殘差塊，Self-Attention，譜歸一化(Spectral Normalization)等等
• torchgan.logging：提供了很強(qiáng)的可視化工具接口，包括對損失函數(shù)、梯度、測量標(biāo)準(zhǔn)以及生成圖片的可視化等
• torchgan.losses：常見的訓(xùn)練 GANs 模型的損失函數(shù)，包括原始的對抗損失、最小二乘損失、WGAN的損失函數(shù)等；
• torchgan.metrics：主要是提供了不同的評(píng)判測量標(biāo)準(zhǔn)
• torchgan.models：包含常見的 GAN 網(wǎng)絡(luò)結(jié)構(gòu)，可以直接使用并且也可以進(jìn)行拓展，包括 DCGAN、cGAN等
• torchgan.trainer：主要是提供訓(xùn)練模型的函數(shù)接口

教程

教程部分如下所示：

教程給出了幾個(gè)例子，包括 DCGAN、Self-Attention GAN、CycleGAN 例子，以及如何自定義損傷的方法。

對于 Self-Attention GAN，還提供了一個(gè)在谷歌的 Colab 運(yùn)行的例子，查看鏈接：

https://torchgan.readthedocs.io/en/latest/tutorials/sagan.html

---

小結(jié)

最后，再給出 Github 項(xiàng)目的鏈接和文檔的對應(yīng)鏈接地址：

https://github.com/torchgan/torchgan

https://torchgan.readthedocs.io/en/latest/index.html

歡迎關(guān)注我的微信公眾號(hào)–機(jī)器學(xué)習(xí)與計(jì)算機(jī)視覺，或者掃描下方的二維碼，大家一起交流，學(xué)習(xí)和進(jìn)步！

之前分享的資源和教程文章有：

• 推薦幾本數(shù)據(jù)結(jié)構(gòu)算法書籍和課程
• [資源分享] Github上八千Star的深度學(xué)習(xí)500問教程
• [資源分享] 吳恩達(dá)最新《機(jī)器學(xué)習(xí)訓(xùn)練秘籍》中文版可以免費(fèi)下載了！
• [資源分享] TensorFlow 官方中文版教程來了
• 必讀的AI和深度學(xué)習(xí)博客
• [教程]一份簡單易懂的 TensorFlow 教程
• 谷歌開源的 GAN 庫–TFGAN

總結(jié)

以上是生活随笔為你收集整理的[资源]基于 Pytorch 的 TorchGAN开源了！的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。