| | |
| | from mmseg.registry import DATASETS |
| | from .basesegdataset import BaseSegDataset |
| |
|
| |
|
| | @DATASETS.register_module() |
| | class MapillaryDataset_v1(BaseSegDataset): |
| | """Mapillary Vistas Dataset. |
| | |
| | Dataset paper link: |
| | http://ieeexplore.ieee.org/document/8237796/ |
| | |
| | v1.2 contain 66 object classes. |
| | (37 instance-specific) |
| | |
| | v2.0 contain 124 object classes. |
| | (70 instance-specific, 46 stuff, 8 void or crowd). |
| | |
| | The ``img_suffix`` is fixed to '.jpg' and ``seg_map_suffix`` is |
| | fixed to '.png' for Mapillary Vistas Dataset. |
| | """ |
| | METAINFO = dict( |
| | classes=('Bird', 'Ground Animal', 'Curb', 'Fence', 'Guard Rail', |
| | 'Barrier', 'Wall', 'Bike Lane', 'Crosswalk - Plain', |
| | 'Curb Cut', 'Parking', 'Pedestrian Area', 'Rail Track', |
| | 'Road', 'Service Lane', 'Sidewalk', 'Bridge', 'Building', |
| | 'Tunnel', 'Person', 'Bicyclist', 'Motorcyclist', |
| | 'Other Rider', 'Lane Marking - Crosswalk', |
| | 'Lane Marking - General', 'Mountain', 'Sand', 'Sky', 'Snow', |
| | 'Terrain', 'Vegetation', 'Water', 'Banner', 'Bench', |
| | 'Bike Rack', 'Billboard', 'Catch Basin', 'CCTV Camera', |
| | 'Fire Hydrant', 'Junction Box', 'Mailbox', 'Manhole', |
| | 'Phone Booth', 'Pothole', 'Street Light', 'Pole', |
| | 'Traffic Sign Frame', 'Utility Pole', 'Traffic Light', |
| | 'Traffic Sign (Back)', 'Traffic Sign (Front)', 'Trash Can', |
| | 'Bicycle', 'Boat', 'Bus', 'Car', 'Caravan', 'Motorcycle', |
| | 'On Rails', 'Other Vehicle', 'Trailer', 'Truck', |
| | 'Wheeled Slow', 'Car Mount', 'Ego Vehicle', 'Unlabeled'), |
| | palette=[[165, 42, 42], [0, 192, 0], [196, 196, 196], [190, 153, 153], |
| | [180, 165, 180], [90, 120, 150], [102, 102, 156], |
| | [128, 64, 255], [140, 140, 200], [170, 170, 170], |
| | [250, 170, 160], [96, 96, 96], |
| | [230, 150, 140], [128, 64, 128], [110, 110, 110], |
| | [244, 35, 232], [150, 100, 100], [70, 70, 70], [150, 120, 90], |
| | [220, 20, 60], [255, 0, 0], [255, 0, 100], [255, 0, 200], |
| | [200, 128, 128], [255, 255, 255], [64, 170, |
| | 64], [230, 160, 50], |
| | [70, 130, 180], [190, 255, 255], [152, 251, 152], |
| | [107, 142, 35], [0, 170, 30], [255, 255, 128], [250, 0, 30], |
| | [100, 140, 180], [220, 220, 220], [220, 128, 128], |
| | [222, 40, 40], [100, 170, 30], [40, 40, 40], [33, 33, 33], |
| | [100, 128, 160], [142, 0, 0], [70, 100, 150], [210, 170, 100], |
| | [153, 153, 153], [128, 128, 128], [0, 0, 80], [250, 170, 30], |
| | [192, 192, 192], [220, 220, 0], [140, 140, 20], [119, 11, 32], |
| | [150, 0, 255], [0, 60, 100], [0, 0, 142], [0, 0, 90], |
| | [0, 0, 230], [0, 80, 100], [128, 64, 64], [0, 0, 110], |
| | [0, 0, 70], [0, 0, 192], [32, 32, 32], [120, 10, |
| | 10], [0, 0, 0]]) |
| |
|
| | def __init__(self, |
| | img_suffix='.jpg', |
| | seg_map_suffix='.png', |
| | **kwargs) -> None: |
| | super().__init__( |
| | img_suffix=img_suffix, seg_map_suffix=seg_map_suffix, **kwargs) |
| |
|
| |
|
| | @DATASETS.register_module() |
| | class MapillaryDataset_v2(BaseSegDataset): |
| | """Mapillary Vistas Dataset. |
| | |
| | Dataset paper link: |
| | http://ieeexplore.ieee.org/document/8237796/ |
| | |
| | v1.2 contain 66 object classes. |
| | (37 instance-specific) |
| | |
| | v2.0 contain 124 object classes. |
| | (70 instance-specific, 46 stuff, 8 void or crowd). |
| | |
| | The ``img_suffix`` is fixed to '.jpg' and ``seg_map_suffix`` is |
| | fixed to '.png' for Mapillary Vistas Dataset. |
| | """ |
| | METAINFO = dict( |
| | classes=( |
| | 'Bird', 'Ground Animal', 'Ambiguous Barrier', 'Concrete Block', |
| | 'Curb', 'Fence', 'Guard Rail', 'Barrier', 'Road Median', |
| | 'Road Side', 'Lane Separator', 'Temporary Barrier', 'Wall', |
| | 'Bike Lane', 'Crosswalk - Plain', 'Curb Cut', 'Driveway', |
| | 'Parking', 'Parking Aisle', 'Pedestrian Area', 'Rail Track', |
| | 'Road', 'Road Shoulder', 'Service Lane', 'Sidewalk', |
| | 'Traffic Island', 'Bridge', 'Building', 'Garage', 'Tunnel', |
| | 'Person', 'Person Group', 'Bicyclist', 'Motorcyclist', |
| | 'Other Rider', 'Lane Marking - Dashed Line', |
| | 'Lane Marking - Straight Line', 'Lane Marking - Zigzag Line', |
| | 'Lane Marking - Ambiguous', 'Lane Marking - Arrow (Left)', |
| | 'Lane Marking - Arrow (Other)', 'Lane Marking - Arrow (Right)', |
| | 'Lane Marking - Arrow (Split Left or Straight)', |
| | 'Lane Marking - Arrow (Split Right or Straight)', |
| | 'Lane Marking - Arrow (Straight)', 'Lane Marking - Crosswalk', |
| | 'Lane Marking - Give Way (Row)', |
| | 'Lane Marking - Give Way (Single)', |
| | 'Lane Marking - Hatched (Chevron)', |
| | 'Lane Marking - Hatched (Diagonal)', 'Lane Marking - Other', |
| | 'Lane Marking - Stop Line', 'Lane Marking - Symbol (Bicycle)', |
| | 'Lane Marking - Symbol (Other)', 'Lane Marking - Text', |
| | 'Lane Marking (only) - Dashed Line', |
| | 'Lane Marking (only) - Crosswalk', 'Lane Marking (only) - Other', |
| | 'Lane Marking (only) - Test', 'Mountain', 'Sand', 'Sky', 'Snow', |
| | 'Terrain', 'Vegetation', 'Water', 'Banner', 'Bench', 'Bike Rack', |
| | 'Catch Basin', 'CCTV Camera', 'Fire Hydrant', 'Junction Box', |
| | 'Mailbox', 'Manhole', 'Parking Meter', 'Phone Booth', 'Pothole', |
| | 'Signage - Advertisement', 'Signage - Ambiguous', 'Signage - Back', |
| | 'Signage - Information', 'Signage - Other', 'Signage - Store', |
| | 'Street Light', 'Pole', 'Pole Group', 'Traffic Sign Frame', |
| | 'Utility Pole', 'Traffic Cone', 'Traffic Light - General (Single)', |
| | 'Traffic Light - Pedestrians', 'Traffic Light - General (Upright)', |
| | 'Traffic Light - General (Horizontal)', 'Traffic Light - Cyclists', |
| | 'Traffic Light - Other', 'Traffic Sign - Ambiguous', |
| | 'Traffic Sign (Back)', 'Traffic Sign - Direction (Back)', |
| | 'Traffic Sign - Direction (Front)', 'Traffic Sign (Front)', |
| | 'Traffic Sign - Parking', 'Traffic Sign - Temporary (Back)', |
| | 'Traffic Sign - Temporary (Front)', 'Trash Can', 'Bicycle', 'Boat', |
| | 'Bus', 'Car', 'Caravan', 'Motorcycle', 'On Rails', 'Other Vehicle', |
| | 'Trailer', 'Truck', 'Vehicle Group', 'Wheeled Slow', 'Water Valve', |
| | 'Car Mount', 'Dynamic', 'Ego Vehicle', 'Ground', 'Static', |
| | 'Unlabeled'), |
| | palette=[[165, 42, 42], [0, 192, 0], [250, 170, 31], [250, 170, 32], |
| | [196, 196, 196], [190, 153, 153], [180, 165, 180], |
| | [90, 120, 150], [250, 170, 33], [250, 170, 34], |
| | [128, 128, 128], [250, 170, 35], [102, 102, 156], |
| | [128, 64, 255], [140, 140, 200], [170, 170, 170], |
| | [250, 170, 36], [250, 170, 160], [250, 170, 37], [96, 96, 96], |
| | [230, 150, 140], [128, 64, 128], [110, 110, 110], |
| | [110, 110, 110], [244, 35, 232], [128, 196, |
| | 128], [150, 100, 100], |
| | [70, 70, 70], [150, 150, 150], [150, 120, 90], [220, 20, 60], |
| | [220, 20, 60], [255, 0, 0], [255, 0, 100], [255, 0, 200], |
| | [255, 255, 255], [255, 255, 255], [250, 170, 29], |
| | [250, 170, 28], [250, 170, 26], [250, 170, |
| | 25], [250, 170, 24], |
| | [250, 170, 22], [250, 170, 21], [250, 170, |
| | 20], [255, 255, 255], |
| | [250, 170, 19], [250, 170, 18], [250, 170, |
| | 12], [250, 170, 11], |
| | [255, 255, 255], [255, 255, 255], [250, 170, 16], |
| | [250, 170, 15], [250, 170, 15], [255, 255, 255], |
| | [255, 255, 255], [255, 255, 255], [255, 255, 255], |
| | [64, 170, 64], [230, 160, 50], |
| | [70, 130, 180], [190, 255, 255], [152, 251, 152], |
| | [107, 142, 35], [0, 170, 30], [255, 255, 128], [250, 0, 30], |
| | [100, 140, 180], [220, 128, 128], [222, 40, |
| | 40], [100, 170, 30], |
| | [40, 40, 40], [33, 33, 33], [100, 128, 160], [20, 20, 255], |
| | [142, 0, 0], [70, 100, 150], [250, 171, 30], [250, 172, 30], |
| | [250, 173, 30], [250, 174, 30], [250, 175, |
| | 30], [250, 176, 30], |
| | [210, 170, 100], [153, 153, 153], [153, 153, 153], |
| | [128, 128, 128], [0, 0, 80], [210, 60, 60], [250, 170, 30], |
| | [250, 170, 30], [250, 170, 30], [250, 170, |
| | 30], [250, 170, 30], |
| | [250, 170, 30], [192, 192, 192], [192, 192, 192], |
| | [192, 192, 192], [220, 220, 0], [220, 220, 0], [0, 0, 196], |
| | [192, 192, 192], [220, 220, 0], [140, 140, 20], [119, 11, 32], |
| | [150, 0, 255], [0, 60, 100], [0, 0, 142], [0, 0, 90], |
| | [0, 0, 230], [0, 80, 100], [128, 64, 64], [0, 0, 110], |
| | [0, 0, 70], [0, 0, 142], [0, 0, 192], [170, 170, 170], |
| | [32, 32, 32], [111, 74, 0], [120, 10, 10], [81, 0, 81], |
| | [111, 111, 0], [0, 0, 0]]) |
| |
|
| | def __init__(self, |
| | img_suffix='.jpg', |
| | seg_map_suffix='.png', |
| | **kwargs) -> None: |
| | super().__init__( |
| | img_suffix=img_suffix, seg_map_suffix=seg_map_suffix, **kwargs) |
| |
|
