| | |
| | import mmengine.fileio as fileio |
| |
|
| | from mmseg.registry import DATASETS |
| | from .basesegdataset import BaseSegDataset |
| |
|
| |
|
| | @DATASETS.register_module() |
| | class iSAIDDataset(BaseSegDataset): |
| | """ iSAID: A Large-scale Dataset for Instance Segmentation in Aerial Images |
| | In segmentation map annotation for iSAID dataset, which is included |
| | in 16 categories. ``reduce_zero_label`` is fixed to False. The |
| | ``img_suffix`` is fixed to '.png' and ``seg_map_suffix`` is fixed to |
| | '_manual1.png'. |
| | """ |
| |
|
| | METAINFO = dict( |
| | classes=('background', 'ship', 'store_tank', 'baseball_diamond', |
| | 'tennis_court', 'basketball_court', 'Ground_Track_Field', |
| | 'Bridge', 'Large_Vehicle', 'Small_Vehicle', 'Helicopter', |
| | 'Swimming_pool', 'Roundabout', 'Soccer_ball_field', 'plane', |
| | 'Harbor'), |
| | palette=[[0, 0, 0], [0, 0, 63], [0, 63, 63], [0, 63, 0], [0, 63, 127], |
| | [0, 63, 191], [0, 63, 255], [0, 127, 63], [0, 127, 127], |
| | [0, 0, 127], [0, 0, 191], [0, 0, 255], [0, 191, 127], |
| | [0, 127, 191], [0, 127, 255], [0, 100, 155]]) |
| |
|
| | def __init__(self, |
| | img_suffix='.png', |
| | seg_map_suffix='_instance_color_RGB.png', |
| | ignore_index=255, |
| | **kwargs) -> None: |
| | super().__init__( |
| | img_suffix=img_suffix, |
| | seg_map_suffix=seg_map_suffix, |
| | ignore_index=ignore_index, |
| | **kwargs) |
| | assert fileio.exists( |
| | self.data_prefix['img_path'], backend_args=self.backend_args) |
| |
|
