ktda/datasets/l8_biome.py

from mmseg.registry import DATASETS
from mmseg.datasets import BaseSegDataset


@DATASETS.register_module()
class L8BIOMEDataset(BaseSegDataset):
    METAINFO = dict(
        classes=("Clear", "Cloud Shadow", "Thin Cloud", "Cloud"),
        palette=[
            [79, 253, 199],
            [221, 53, 223],
            [251, 255, 41],
            [77, 2, 115],
        ],
    )

    def __init__(
        self,
        img_suffix=".png",
        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
        )