chainer implementation of pix2pix
chainer implementation of pix2pix https://phillipi.github.io/pix2pix/
The Japanese readme can be found here.
pip install -r requirements.txt
python train_facade.py -g [GPU ID, e.g. 0] -i [dataset root directory] --out [output directory] --snapshot_interval 10000
--out
stores snapshots of the model and example images at an interval defined by --snapshot_interval
--snapshot_interval
to save resources.facade_dataset.py
for your dataset. The function get_example should be written so that it returns the i-th image pair a tuple of numpy arrays i.e. (input, output)
.updater.py
.facade_visualizer.py
and modify to visualize the dataset.train_facade.py
change in_ch
and out_ch
to the correct input and output channels for your data.