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

Computes and stores the average and current value

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

Defines a PatchGAN discriminator

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

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)

Define a Resnet block

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 is a subclass of Pix2PixModel for image colorization (black & white image -> colorful images). The model training requires '-dataset_model c

before coming to use functions of this class, the lanes should resize to im_width and im_height using resize_lane() in lane_compare.hpp

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

Randomly horizontally flips the given numpy Image with a probability of 0.5

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

A template dataset class for you to implement custom datasets.

This TesteModel can be used to generate CycleGAN results for only one direction. This model will automatically set '--dataset_mode single', which

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