1. 程式人生 > >“史上最強”BigGAN公開TensorFlow Hub demo!

“史上最強”BigGAN公開TensorFlow Hub demo!

還記得前些日子轟動一時的 BigGAN 模型嗎?生成對抗網路(GAN)作為當前最熱門的技術之一,最近在影象生成方面的成果頗受人關注。近日,由 DeepMind 和赫瑞瓦特大學組成的科研人員公佈的 BigGAN 模型,被稱為“史上最強 GAN 生成器”,可合成食物照片、風景肖像和動物照片,其圖片效果十分逼真,在 ImageNet 的測評集上從之前的最高分 52.52 提升到 66.3,效果提升非常顯著。


640?wx_fmt=png

由BigGAN模型生成的類條件樣本(圖片擷取自論文)


目前 ICLR 2019 的論文評審結果於上週剛剛公佈,其中就包括 BigGAN 模型的這篇論文“Large Scale GAN Training for High Fidelity Natural Image Synthesis”,現已掛在 arXiv 上。BigGAN 獲得了三位評審 8 分、7 分和 10 分的評價,詳情參看:

https://openreview.net/forum?id=B1xsqj09Fm


簡單解釋下,之所以稱為 BigGAN,引用合肥工業大學研究生武廣的話來講,主要歸於三點:


一是 Batch Size 的增大,二是採用先驗分佈 z 的“截斷技巧”,三是對模型穩定性的控制。


如在訓練中每批 Batch 就有 2048 張照片(通常情況下都是 64 張),用於控制模型部分特性的演算法控制器引數是此前其他模型的 4 倍,近 1.58 億。


為方便開發者練習和使用,DeepMind 今天又公開了 BigGAN 的 TensorFlow Hub,順帶還提供了在 Colab 上的 demo,開發者可以在 Colab 上執行影象生成和插值任務。


連結:https://tfhub.dev/deepmind/biggan-256


640?wx_fmt=png


如何使用 Colab demo?


首先,你得開啟 Colab,登入  Google 賬戶。


注意:這是一個完全執行在雲端的環境,開發者可以使用 Keras、TensorFlow、PyTorch 等框架進行深度學習應用的開發,尤其適合那些手頭沒有 GPU 資源的初學者。


第一步:設定


640?wx_fmt=png


1.1 從 TF Hub 中下載 BigGAN 生成器模組


640?wx_fmt=png


1.2 定義 BigGAN 影象分類及展示功能


640?wx_fmt=jpeg


1.3 建立 TensorFlow 會話,並初始化變數


640?wx_fmt=png


第二步:搜尋特定類別的 BigGAN 樣本


  • 類條件抽樣


640?wx_fmt=png


第三步:在 BigGAN 樣本之間插值


  • 插值


640?wx_fmt=png


參考連結:

Colab demo:

https://colab.research.google.com/github/tensorflow/hub/blob/master/examples/colab/biggan_generation_with_tf_hub.ipynb


——【完】——


2018 中國大資料技術大會將於 12 月 6 - 8 日在新雲南皇冠假日酒店舉行。匯聚超百位國內外實力講師從學界翹楚到行業一線大拿:

管曉巨集:中國科學院院士;

張巨集江:原始碼資本投資合夥人;

張曉東:美國俄亥俄州立大學 Robert M. Critchfield 講席教授;

陳性元:北京資訊科學技術研究院副院長;

周靖人:阿里巴巴集團副總裁;

李浩源:Alluxio 公司創始人&CEO

......

全方位立體解讀大資料時代的技術程序,為眾技術愛好者奉上一場優質乾貨盛宴。


640?wx_fmt=jpeg