<filename>data_loader/__init__.py
from .cityscapes import CitySegmentation
from .forest import Forest

datasets = {
    'citys': CitySegmentation,
    'forest': Forest,
}


def get_segmentation_dataset(name, **kwargs):
    """Segmentation Datasets"""
    return datasets[name.lower()](**kwargs)

<filename>data_loader/__init__.py
from .cityscapes import CitySegmentation
from .forest import Forest

datasets = {
    'citys': CitySegmentation,
    'forest': Forest,
}


def get_segmentation_dataset(name, **kwargs):
    """Segmentation Datasets"""
    return datasets[name.lower()](**kwargs)

Segmentation Datasets