trw.datasets.cyclegan

Module Contents

Functions

image_to_torch(i)

load_case(batch, dataset_a, dataset_b, transform, aligned)

train_transform()

create_cycle_gan_dataset(dataset_name: cycle_gan_dataset, batch_size: int = 32, root: Optional[str] = None, url: str = 'https://people.eecs.berkeley.edu/~taesung_park/CycleGAN/datasets/', transform_train: Optional[List[trw.transforms.Transform]] = None, transform_valid: Optional[List[trw.transforms.Transform]] = None, aligned: bool = False, loading_batch_size: int = 4, nb_workers: int = 4) → trw.basic_typing.Datasets

Datasets used for image to image translation (domain A to domain B).

Attributes

cycle_gan_dataset

trw.datasets.cyclegan.cycle_gan_dataset
trw.datasets.cyclegan.image_to_torch(i)
trw.datasets.cyclegan.load_case(batch, dataset_a, dataset_b, transform, aligned)
trw.datasets.cyclegan.train_transform()
trw.datasets.cyclegan.create_cycle_gan_dataset(dataset_name: cycle_gan_dataset, batch_size: int = 32, root: Optional[str] = None, url: str = 'https://people.eecs.berkeley.edu/~taesung_park/CycleGAN/datasets/', transform_train: Optional[List[trw.transforms.Transform]] = None, transform_valid: Optional[List[trw.transforms.Transform]] = None, aligned: bool = False, loading_batch_size: int = 4, nb_workers: int = 4) → trw.basic_typing.Datasets

Datasets used for image to image translation (domain A to domain B).

Parameters

  • dataset_name – the name of the dataset

  • batch_size – the size of each batch

  • root – the root path where to store the dataset

  • url – specify the URL from which the dataset is downloaded

  • transform_train – transform applied to train dataset

  • transform_valid – transform applied to valid dataset

  • aligned – if True, the images A and B will be considered aligned. If False, B will be randomly sampled from the list of available images in B

  • nb_workers – the number of workers to process the images

  • loading_batch_size – the number of images loaded by a worker

Returns

a dataset