simulation.utils.machine_learning.cycle_gan.test module

Summary

Functions:

test_on_dataset

Test one dataset and save all images.

Reference

test_on_dataset(dataset: simulation.utils.machine_learning.data.data_loader.DataLoader, generators: Tuple[torch.nn.modules.module.Module, torch.nn.modules.module.Module], class_names: Tuple[str, str], destination: str, aspect_ratio: float = 1, device: torch.device = device(type='cpu'))[source]

Test one dataset and save all images.

Parameters

  • dataset – The dataset to test

  • generators – Both generators. Second one is used to generate the fake image.

  • class_names – The class names to save the images correctly.

  • destination – The destination folder

  • aspect_ratio – The aspect ratio of the images

  • device – the device on which the models are located