Source code for simulation.utils.machine_learning.cycle_gan.configs.test_options

from typing import List

from .base_options import BaseOptions


class TestOptions(BaseOptions):
    dataset_a: List[str] = ["./../../../../data/real_images/maschinen_halle_parking"]
    """Path to images of domain A (real images)."""
    dataset_b: List[str] = ["./../../../../data/simulated_images/test_images"]
    """Path to images of domain B (simulated images)."""
    results_dir: str = "./results/"
    """Saves results here."""
    aspect_ratio: float = 1
    """Aspect ratio of result images."""
    is_train: bool = False
    """Enable or disable training mode."""


class WassersteinCycleGANTestOptions(TestOptions):
    pass


class CycleGANTestOptions(TestOptions):
    pass