File size: 717 Bytes
918db92 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
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
) |