Image Coloring GAN

• 取り組んだ期間: 4ヶ月弱
• 使用技術
  • Python, PyTorch, pix2pix (GAN)
• 内容
  • image-to-image タスクで高い性能を発揮していた GAN ベースの pix2pix を用いて、白黒画像の自動着色モデルを学習した。
    
  • Kaggle で公開されている風景画像のデータセットをグレースケール化して 4,300枚ほどのデータセットを作成し、訓練用、検証用、評価テスト用データに 8:1:1 で分割して使用した。

実験に使ったコードなどは こちらのリポジトリにあります。