def __init__(self, mode='fine', **kwargs):
self.num_classes = 19
self.mode = mode
self.palette = palette.create_palette('cityscapes')
self.id_to_trainId = ID_TO_TRAINID
super(CityScapesDataset, self).__init__(data_name='cityscapes',
**kwargs)
def __init__(self, **kwargs):
self.num_classes = 2
self.sequence_num = 4
self.palette = palette.create_palette('lanesequence')
super(RobustDataset, self).__init__(data_name='lanesequence', **kwargs)
def __init__(self, save_exist=False, **kwargs):
self.num_classes = 5
self.save_exist = save_exist
self.palette = palette.create_palette('culane')
super(CULaneDataset, self).__init__(data_name='culane', **kwargs)
def __init__(self, **kwargs):
self.num_classes = 2
self.palette = palette.create_palette('lanesequence')
super(TuSimpleDataset, self).__init__(data_name='tusimple', **kwargs)
def __init__(self, **kwargs):
self.palette = palette.create_palette('voc')
self.num_classes = 21
super(VOCDataset, self).__init__(data_name='voc', **kwargs)
def __init__(self, number_classes, **kwargs):
self.number_classes = number_classes
self.palette = palette.create_palette(name=None,
cls_num=self.number_classes)
super(CocoStandard, self).__init__(data_name='coco', **kwargs)
def __init__(self, **kwargs):
self.num_classes = 182
self.palette = palette.create_palette('coco')
super(CocoStuff10K, self).__init__(data_name='cocoStuff10k', **kwargs)