This class implements an image buffer that stores previously generated images. This buffer enables us to update discriminators using a history of

Defines the Unet submodule with skip connection. X -------------------identity---------------------- |-- downsampling -- |submodule| -

Define a Resnet block

Resnet-based generator that consists of Resnet blocks between a few downsampling/upsampling operations. We adapt Torch code and idea from Justin

Defines a PatchGAN discriminator

Create a Unet-based generator

Wrapper class of Dataset class that performs multi-threaded data loading

This HTML class allows us to save images and write texts into a single HTML file. It consists of functions such as <add_header> (add a text head

Defines a 1x1 PatchGAN discriminator (pixelGAN)

This class includes test options. It also includes shared options defined in BaseOptions.

This class includes training options. It also includes shared options defined in BaseOptions.

This class includes several functions that can display/save images and print/save logging information. It uses a Python library 'visdom' for disp

This class is an abstract base class (ABC) for datasets. To create a subclass, you need to implement the following four functions: -- <__init

This class is an abstract base class (ABC) for models. To create a subclass, you need to implement the following five functions: -- <__ini

This class defines options used during both training and test time. It also implements several helper functions such as parsing, printing, and sa

This class implements CUT and FastCUT model, described in the paper Contrastive Learning for Unpaired Image-to-Image Translation Taesung Park

This class implements the CycleGAN model, for learning image-to-image translation without paired data. The model training requires '--datas

This class implements DCLGAN model. This code is inspired by CUT and CycleGAN.

This class implements CUT and FastCUT model, described in the paper Contrastive Learning for Unpaired Image-to-Image Translation Taesung Park

Define different GAN objectives. The GANLoss class abstracts away the need to create the target label tensor that has the same size as the in

A Python script for downloading CycleGAN or pix2pix datasets. Parameters: technique (str) -- One of: 'cyclegan' or 'pix2pix'. ver

Defines a PatchGAN discriminator

Resnet-based decoder that consists of a few Resnet blocks + a few upsampling operations.

Resnet-based encoder that consists of a few downsampling + several Resnet blocks

This class implements DCLGAN with similarity loss. This code is inspired by CUT and CycleGAN.

This dataset class can load a set of images specified by the path --dataroot /path/to/data. It can be used for generating CycleGAN results only f

This dataset class can load unaligned/unpaired datasets. It requires two directories to host training images from domain A '/path/to/data/tr

A template dataset class for you to implement custom datasets.

This dataset class can load unaligned/unpaired datasets. It requires two directories to host training images from domain A '/path/to/data/tr