Graph2Pix: A Graph-Based Image to Image Translation Framework

Sep 13, 2021 1 min read

Graph2Pix: A Graph-Based Image to Image Translation Framework

Installation

Install the dependencies in env.yml

$ conda env create -f env.yml
$ conda activate pix-env

Examples

Train ArtBreeder dataset with Smart Disc Feature enabled and 7 disc images.

$ python train.py --dataroot ./datasets/newbreeder/ --name art_newbreeder_allparents_bs8_disc --no_instance --fineSize 256 --loadSize 256 --label_nc 0 --resize_or_crop resize_and_crop --output_nc 3 --batchSize 8 --mv --smart_disc --num_disc_images 7

Generate images from trained model.

$ python gen_imgs.py --dataroot ./datasets/newbreeder/ --name art_newbreeder_allparents_bs8_disc --no_instance --fineSize 256 --loadSize 256 --label_nc 0 --resize_or_crop resize_and_crop --output_nc 3 --batchSize 8 --mv --smart_disc --num_disc_images 7 --which_epoch 50

FID Calculation

$ python -m pytorch_fid gen_images/art_newbreeder_allparents_bs8_disc/gts gen_images/art_newbreeder_allparents_bs8_disc/generated/

Acknowledgments

The basis of this code is pix2pixHD.

GitHub

John was the first writer to have joined pythonawesome.com. He has since then inculcated very effective writing and reviewing culture at pythonawesome which rivals have found impossible to imitate.