GAN 自從被提出後，便迅速受到廣泛關注。我們可以將 GAN 分為兩類，一類是無條件下的生成；另一類是基於條件信息的生成。近日，來自韓國浦項科技大學的碩士生在 GitHub 上開源了一個項目，提供了條件 / 無條件圖像生成的代表性生成對抗網絡（GAN）的實現。
近日，機器之心在 GitHub 上看到了一個非常有意義的項目 PyTorch-StudioGAN，它是一個 PyTorch 庫，提供了條件 / 無條件圖像生成的代表性生成對抗網絡（GAN）的實現。據主頁介紹，該項目旨在提供一個統一的現代 GAN 平臺，這樣機器學習領域的研究者可以快速地比較和分析新思路和新方法等。該項目的作者為韓國浦項科技大學的碩士生，他的研究興趣主要包括深度學習、機器學習和計算機視覺。項目地址：https://github.com/POSTECH-CVLab/PyTorch-StudioGAN提供了大量 PyTorch 框架的 GAN 實現；
基於 CIFAR 10、Tiny ImageNet 和 ImageNet 數據集的 GAN 基準；
相較原始實現的更好的性能和更低的內存消耗；
提供完全最新 PyTorch 環境的預訓練模型；
支持多 GPU（DP、DDP 和多節點 DDP）、混合精度、同步批歸一化、LARS、Tensorboard 可視化和其他分析方法。
對於這個 PyTorch GAN 庫，有網友表示：「看上去很不錯！如果可以提供 top-k 等現代訓練實踐以及各種增強方法就更棒了。」對此，項目作者稱其會在 NeurIPS 論文提交截止日期之後，添加一些改進的方法，如 Sinha 等人的 Tok-K 訓練以及 Langevin 採樣和 SimCLR 增強。此外，有網友詢問是否可以將該項目用於圖像之外的其他領域。作者表示可以，即使無法使用一些穩定器（如 diffaug、ada 等），依然可以通過調整 dataLoader 來訓練自己的模型。如下圖所示，項目作者提供了 18 + 個 SOTA GAN 的實現，包括 DCGAN、LSGAN、GGAN、WGAN-WC、WGAN-GP、WGAN-DRA、ACGAN、ProjGAN、SNGAN、SAGAN、BigGAN、BigGAN-Deep、CRGAN、ICRGAN、LOGAN、DiffAugGAN、ADAGAN、ContraGAN 和 FreezeD。cBN：條件批歸一化；AC：輔助分類器；PD：Projection 判別器；CL：對比學習。下圖中 StyleGAN2 為即將實現的 GAN 網絡，其中 AdaIN 表示自適應實例歸一化（Adaptive Instance Normalization）。Anaconda
Python >= 3.6
6.0.0 <= Pillow <= 7.0.0
scipy == 1.1.0
sklearn
seaborn
h5py
tqdm
torch >= 1.6.0
torchvision >= 0.7.0
tensorboard
5.4.0 <= gcc <= 7.4.0
torchlars
conda env create -f environment.yml -n studiogandocker pull mgkang/studiogan:latest以下是創建名字為「studioGAN」容器的命令，同樣也可以使用埠號為 6006 來連接 tensoreboard。docker run -it --gpus all --shm-size 128g -p 6006:6006 --name studioGAN -v /home/USER:/root/code --workdir /root/code mgkang/studiogan:latest /bin/bash使用 GPU 0 的情況下，在 CONFIG_PATH 中對於模型的訓練「-t」和評估「-e」進行了定義：CUDA_VISIBLE_DEVICES=0 python3 src/main.py -t -e -c CONFIG_PATH在使用 GPU (0, 1, 2, 3) 和 DataParallel 情況下，在 CONFIG_PATH 中對於模型的訓練「-t」和評估「-e」進行了定義：CUDA_VISIBLE_DEVICES=0,1,2,3 python3 src/main.py -t -e -c CONFIG_PATH在 python3 src/main.py 程序中查看可用選項，通過 Tensorboard 可以監控 IS、FID、F_beta、Authenticity Accuracies 以及最大奇異值：~ PyTorch-StudioGAN/logs/RUN_NAME>>> tensorboard --logdir=./ --port PORTStudioGAN 支持圖像可視化、k 最近鄰分析、線性差值以及頻率分析。所有的結果保存在「./figures/RUN_NAME/*.png」中。CUDA_VISIBLE_DEVICES=0,...,N python3 src/main.py -iv -std_stat --standing_step STANDING_STEP -c CONFIG_PATH --checkpoint_folder CHECKPOINT_FOLDER --log_output_path LOG_OUTPUT_PATHk 最近鄰分析，這裡固定 K=7，第一列中是生成的圖像：CUDA_VISIBLE_DEVICES=0,...,N python3 src/main.py -knn -std_stat --standing_step STANDING_STEP -c CONFIG_PATH --checkpoint_folder CHECKPOINT_FOLDER --log_output_path LOG_OUTPUT_PATH線性插值（僅適用於有條件的 Big ResNet 模型 ）的代碼和示例如下：CUDA_VISIBLE_DEVICES=0,...,N python3 src/main.py -itp -std_stat --standing_step STANDING_STEP -c CONFIG_PATH --checkpoint_folder CHECKPOINT_FOLDER --log_output_path LOG_OUTPUT_PATH參考連結：https://www.reddit.com/r/MachineLearning/comments/lu9gen/p_pytorch_gan_library_that_provides/AWS白皮書《策略手冊：數據、 分析與機器學習》
曾存儲過 GB 級業務數據的組織現在發現，所存儲的數據量現已達 PB 級甚至 EB 級。要充分利用這 些海量數據的價值，就需要利用現代化雲數據基礎設施，從而將不同的信息豎井融合統一。無論您處於數據現代化改造過程中的哪個階段，本行動手冊都能幫助您完善策略，在整個企業範圍內高效擴展數據、分析和機器學習，從而加快創新並推動業務發展。點擊閱讀原文，免費領取白皮書。
© THE END
轉載請聯繫本公眾號獲得授權
投稿或尋求報導：content@jiqizhixin.com