<reponame>MSiam/video_class_agnostic_segmentation
from .coco import CocoDataset
from .registry import DATASETS


@DATASETS.register_module
class CityscapesDataset(CocoDataset):

    CLASSES = ("person", "rider", "car", "truck", "bicycle", "motorcycle", "bus", "train")

from .coco import CocoDataset
from .registry import DATASETS


@DATASETS.register_module
class CityscapesDataset(CocoDataset):

    CLASSES = ("person", "rider", "car", "truck", "bicycle", "motorcycle", "bus", "train")