patchGAN image segmentation model in PyTorch
Project description
patchGAN
UNet-based GAN model for image segmentation using a patch-wise discriminator. Based on the pix2pix model.
Installation
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
Training
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.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
patchGAN-0.2.tar.gz
(23.4 kB
view hashes)
Built Distribution
patchGAN-0.2-py3-none-any.whl
(26.3 kB
view hashes)