pix2pix-tf2-keras
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