はじめに
https://github.com/eriklindernoren/PyTorch-GAN
に、沢山のGANがある。
このgithubのコードは、非常に簡単に動作させることができる。
CycleGANを動かしてみた
以下が基本コマンド。
python3 cyclegan.py --dataset_name monet2photo
メモリが足りないと出る。
通常、まず、batch_sizeを小さくしたりするが、
既に、1になっているようなので、
仕方なく、勘で、画像サイズを小さく指定したら、動いてくれた。
--img_height 128 --img_width 128
動作結果
非力なPCなので、epochを極端に小さくした。2にした。
それでも、1日弱時間がかかった。
上段がmonet、下段がそれを入力とした実写(化)、のはず。
まだ、どっちがどっちというレベルではありますが。。。
別途、もう少し、わかりやすいサンプルでやってみます。
コメント
特にありません。
このgithubは、お勧めーーーー。