| | |
| |
|
| | from mmseg.datasets.basesegdataset import BaseSegDataset |
| | from mmseg.registry import DATASETS |
| |
|
| |
|
| | @DATASETS.register_module() |
| | 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) |
| |
|
