UNet-based GAN model for image segmentation using a patch-wise discriminator. Based on the pix2pix model.
Install the package with pip:
pip install patchgan
Upgrading existing install:
pip install -U patchgan
Get the current development branch:
pip install -U git+https://github.com/ramanakumars/patchGAN.git
You can train the patchGAN model with a config file and the patchgan_train command:
patchgan_train --config_file train_coco.yaml --n_epochs 100 --batch_size 16
See examples/train_coco.yaml for the corresponding config for the COCO stuff dataset.