haru-256 / pix2pixHD.pytorch

pix2pixHD.pytorch

本リポジトリはpix2pixHDの論文から，実装をしてみました．

動機

Deep Learnig 系の論文では，「論文で書いてある事と実装でされていることが違っていることがよくある」という背景があります． 実際に実装をしてみると，違いが分かり，論文を読むだけでは得られない知見が得られると考えています．そのため，自分でも再実装してみることにしました．

要件

• Python : 3.6.5
• matplotlib :
• numpy :
• pytorch : 1.04
• opencv : 4.0
• cuda, cuDNN

使用方法

環境構築

• パッケージ関連のインストール

pip install -r requirements

• データセット

学習

テスト

論文との差異

• GAN ロスの部分に WGAN-gp が使えるようにした．
• 最適化手法に AdaBound を選択できるようにした．

参考文献

• 論文：https://arxiv.org/pdf/1711.11585.pdf
• 実装：https://github.com/NVIDIA/pix2pixHD