/pix2pix-tf2-keras

Pix2Pix made using Keras with TensorFlow 2 as backend

Primary LanguagePythonMIT LicenseMIT

pix2pix-tf2-keras

Python Version Tensorflow Version Tensorflow Version GitHub Licence

Pix2Pix made using Keras with TensorFlow 2 as backend

Getting started

pip install -r requirements.txt
python train.py --help
python resume.py --help
python dataset2tfrecord.py --help

Dataloader

Convert your dataset into TFRecord file with dataset2tfrecord.py and choose TFRecordDataLoader as dataloader. Converter requires two folders in which related images must have the same names (with any extension!)

or

Make your own dataloader using dataloader/template.py and add it to dataloader/__init__.py.DATALOADERS dictionary for it to show up in training options

Resuming training

Model creates separate folders for each run in result/ folder. Each run has its config.json, if you are not sure what you are changing there, don't. To resume training from the last checkpoint pass run path (not the config path!) to resume.py

Custom generators

Make your own generator using generator/template.py and add it to generator/__init__.py.GENERATORS dictionary for it to show up in training options