谷歌剛剛宣布升級了新版TF-GAN，這是TensorFlow 2.0的輕量級GAN庫。此次升級和新功能包括：云TPU支持、用于自學(xué)GAN的課程、GAN的衡量指標、無痛pip安裝TF-GAN等。

2017年谷歌推出了TF-GAN。這是一個用于訓(xùn)練和評估生成對抗網(wǎng)絡(luò)(GAN)的輕量級工具庫，已在GitHub開源。

TF-GAN為開發(fā)者提供了輕松訓(xùn)練 GAN 的基礎(chǔ)條件、經(jīng)過完整測試的損失函數(shù)和評估指標，以及易于使用的范例，受到廣泛的好評。

TF-GAN中的訓(xùn)練通常包括以下步驟：

指定網(wǎng)絡(luò)的輸入

使用GANModel設(shè)置生成器和鑒別器

使用GANLoss指定損失

使用GANTrainOps創(chuàng)建訓(xùn)練操作

開始訓(xùn)練模型

而就在今天，谷歌宣布了TF-GAN的升級版。下面我們就來看看都有哪些升級和新功能。

新版更新及新增功能介紹

云TPU支持

開發(fā)者現(xiàn)在可以使用TF-GAN在谷歌的云TPU上訓(xùn)練GAN。TPU是谷歌定制開發(fā)的專用集成電路（ASIC），用于加速機器學(xué)習(xí)工作負載。在其他硬件平臺上需要花費數(shù)周時間進行訓(xùn)練的模型，在TPU上可能只需要數(shù)小時即可完成。

下面這個開源示例演示了如何利用TPU，通過ImageNet訓(xùn)練圖像生成GAN：

https://github.com/tensorflow/gan/tree/master/tensorflow_gan/examples/self_attention_estimator

開發(fā)者還可以免費使用colaboratory，在TPU教程中運行TF-GAN：

https://colab.research.google.com/github/tensorflow/gan/blob/master/tensorflow_gan/examples/colab_notebooks/tfgan_on_tpus.ipynb

用于自學(xué)GAN的課程

當知識能夠免費提供時，機器學(xué)習(xí)效果最佳。為此，谷歌發(fā)布了一個自學(xué)GAN的課程，該課程基于谷歌內(nèi)部已經(jīng)教授多年的GAN課程。

開發(fā)者只需要觀看視頻，閱讀說明，進行練習(xí)；執(zhí)行代碼示例是精通機器學(xué)習(xí)的好方法。

GAN指標

學(xué)術(shù)論文有時會“發(fā)明一個標桿”，然后用它來衡量其他的結(jié)果。

為了便于比較論文的結(jié)果，TF-GAN使得使用標準指標變得更加容易。除了能夠糾正某些困擾標準開源實現(xiàn)的數(shù)值精度和統(tǒng)計偏差之外，TF-GAN指標在計算上是高效的，并且在語法上易于使用。

添加更多范例供開發(fā)者學(xué)習(xí)

GAN研究節(jié)奏更新?lián)Q代特別快。TF-GAN并不打算持續(xù)保留所有GAN模型的實例，不過谷歌還是添加了一些他們覺得比較相關(guān)的內(nèi)容，包括在TPU上訓(xùn)練的Self-Attention GAN。

PyPi包：無痛安裝TF-GAN

TF-GAN現(xiàn)在可以用'pip install tensorflow-gan'安裝，配合'import tensorflow_gan as tfgan'一起服用，效果頗佳。

Colaboratory教程

谷歌改進了之前的教程，現(xiàn)在可以與谷歌的免費GPU和TPU一起使用。

獨立的GitHub倉庫

TF-GAN現(xiàn)在有了自己獨立的GitHub倉庫，便于更輕松地追蹤更改，并正確地為開源貢獻者提供回報。

Github：

https://github.com/tensorflow/gan

兼容TensorFlow 2.0

TF-GAN目前與TF 2.0兼容，但谷歌仍在繼續(xù)使其與Keras兼容。開發(fā)者可以在tensorflow.org/beta找到一些不使用TF-GAN的GAN Keras示例，包括DCGAN，Pix2Pix和CycleGAN。

使用TF-GAN的范例項目

云TPU上的Self-Attention GAN

Self-Attention GAN使用兩個指標，即初始分數(shù)和Frechet初始距離，在圖像生成方面取得了最先進的結(jié)果。谷歌開源了這個模型的兩個版本，其中一個在Cloud TPU上以開源方式運行。TPU版本與GPU版本相同，但訓(xùn)練速度提高了12倍。

下面3張圖片依次為：TPU生成的fake圖片、GPU生成的fake圖片以及真實照片：

圖像擴展

基于周圍環(huán)境填充圖像的缺失部分進行圖像修復(fù)，已經(jīng)是一個研究的很成熟的問題，但相關(guān)的圖像擴展問題的研究較少。

圖像擴展要求算法以“合理且一致”的方式將圖像擴展到其邊界之外。這在虛擬現(xiàn)實環(huán)境中是有用的，其中通常需要模擬不同的相機特性，以及諸如全景拼接之類的計算攝影應(yīng)用，其中不同的圖像需要被平滑地縫合在一起。

谷歌研究工程師最近開發(fā)了一種新算法，該算法使用比以前的方法更少的工件擴展圖像，并使用TPU對其進行訓(xùn)練。

BigGAN

DeepMind研究團隊使用新更改的體系結(jié)構(gòu)、更大的網(wǎng)絡(luò)、更大的batch大小和Google TPU相結(jié)合，改進了本文中最先進的圖像生成。

他們使用TF-GAN的評估模塊來標準化指標，并且能夠展示各種圖像尺寸的質(zhì)量改進。