dataset_configs/CVRP.py

from mmseg.registry import DATASETS
from .basesegdataset import BaseSegDataset

@DATASETS.register_module()
class CVRPDataset(BaseSegDataset):
    METAINFO = {
        'classes':['background','panicle'],
        'palette':[[127,127,127],[200,0,0]]
    }

    def __init__(self,
                 seg_map_suffix='.png',
                 reduce_zero_label=False,
                 **kwargs)->None:
        super().__init__(
            seg_map_suffix=seg_map_suffix,
            reduce_zero_label=reduce_zero_label,
            **kwargs
        )