image_generator_2

概要

メルアイコン生成器 version2で用いたソースコードです。
詳しい解説はこちら。

想定環境

python 3.7.1
pip install -r requirements.txtで環境を揃えることができます。

プログラム

• PGGAN_train.pyは学習を実行し、学習の過程と学習結果を出力するプログラムです。
• PGGAN_inference.pyはPGGAN_train.pyが出力した学習結果をGeneratorに読み込み推論を実行、生成画像を出力するプログラムです。

使い方

1. PGGAN_train.pyのあるディレクトリに./datasetディレクトリを作成します
2. ./datasetディレクトリに、学習に使いたい画像を./dataset/*/*という形式で好きな数入れます(画像のファイル形式はpng)
3. PGGAN_train.pyの置いてあるディレクトリでpython PGGAN_train.pyを実行して学習を開始します
   • 学習の過程が./output_img/pggan_train/以下に出力されます
   • 学習結果がgenerator_trained_model_cpu.pthとして出力されます
4. PGGAN_inference.pyの置いてあるディレクトリでpython PGGAN_inference.pyを実行して推論します
   • 推論結果が./output_img/pggan_inference/以下に出力されます
   • 注意点として、PGGAN_inference.pyの置いてあるディレクトリにgenerator_trained_model_cpu.pthがなければエラーとなります

学習には環境によっては12時間以上要する場合があります。
入力された画像は256×256にリサイズされた上で学習に使われます。出力画像も256×256です。