image_generator

概要

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

想定環境

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

使い方

1. GAN.pyと同じディレクトリに、datasetという名前のディレクトリを作成
2. GAN.pyと同じディレクトリに、imgという名前のディレクトリを作成
3. datasetディレクトリに、学習に使いたい画像を*.jpgという形式で好きな数入れる
4. GAN.pyの置いてあるディレクトリでpython GAN.pyを実行
5. 学習し、その結果を用いて画像を生成、imgディレクトリ以下に生成結果を出力します

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