| |
| from mmseg.registry import DATASETS |
| from .basesegdataset import BaseSegDataset |
|
|
|
|
| @DATASETS.register_module() |
| class LoveDADataset(BaseSegDataset): |
| """LoveDA dataset. |
| |
| In segmentation map annotation for LoveDA, 0 is the ignore index. |
| ``reduce_zero_label`` should be set to True. The ``img_suffix`` and |
| ``seg_map_suffix`` are both fixed to '.png'. |
| """ |
| METAINFO = dict( |
| classes=('background', 'building', 'road', 'water', 'barren', 'forest', |
| 'agricultural'), |
| palette=[[255, 255, 255], [255, 0, 0], [255, 255, 0], [0, 0, 255], |
| [159, 129, 183], [0, 255, 0], [255, 195, 128]]) |
|
|
| def __init__(self, |
| img_suffix='.png', |
| seg_map_suffix='.png', |
| reduce_zero_label=True, |
| **kwargs) -> None: |
| super().__init__( |
| img_suffix=img_suffix, |
| seg_map_suffix=seg_map_suffix, |
| reduce_zero_label=reduce_zero_label, |
| **kwargs) |
|
|
