Spaces:
Runtime error
Runtime error
# Copyright (c) OpenMMLab. All rights reserved. | |
from mmseg.datasets.basesegdataset import BaseSegDataset | |
from mmseg.registry import DATASETS | |
class BDD100KDataset(BaseSegDataset): | |
METAINFO = dict( | |
classes=('road', 'sidewalk', 'building', 'wall', 'fence', 'pole', | |
'traffic light', 'traffic sign', 'vegetation', 'terrain', | |
'sky', 'person', 'rider', 'car', 'truck', 'bus', 'train', | |
'motorcycle', 'bicycle'), | |
palette=[[128, 64, 128], [244, 35, 232], [70, 70, 70], [102, 102, 156], | |
[190, 153, 153], [153, 153, 153], [250, 170, | |
30], [220, 220, 0], | |
[107, 142, 35], [152, 251, 152], [70, 130, 180], | |
[220, 20, 60], [255, 0, 0], [0, 0, 142], [0, 0, 70], | |
[0, 60, 100], [0, 80, 100], [0, 0, 230], [119, 11, 32]]) | |
def __init__(self, | |
img_suffix='.jpg', | |
seg_map_suffix='.png', | |
reduce_zero_label=False, | |
**kwargs) -> None: | |
super().__init__( | |
img_suffix=img_suffix, | |
seg_map_suffix=seg_map_suffix, | |
reduce_zero_label=reduce_zero_label, | |
**kwargs) | |