| | |
| | import os.path as osp |
| |
|
| | import mmengine.fileio as fileio |
| |
|
| | from mmseg.registry import DATASETS |
| | from .basesegdataset import BaseSegDataset |
| |
|
| |
|
| | @DATASETS.register_module() |
| | class PascalVOCDataset(BaseSegDataset): |
| | """Pascal VOC dataset. |
| | |
| | Args: |
| | split (str): Split txt file for Pascal VOC. |
| | """ |
| | METAINFO = dict( |
| | classes=('background', 'aeroplane', 'bicycle', 'bird', 'boat', |
| | 'bottle', 'bus', 'car', 'cat', 'chair', 'cow', 'diningtable', |
| | 'dog', 'horse', 'motorbike', 'person', 'pottedplant', 'sheep', |
| | 'sofa', 'train', 'tvmonitor'), |
| | palette=[[0, 0, 0], [128, 0, 0], [0, 128, 0], [128, 128, 0], |
| | [0, 0, 128], [128, 0, 128], [0, 128, 128], [128, 128, 128], |
| | [64, 0, 0], [192, 0, 0], [64, 128, 0], [192, 128, 0], |
| | [64, 0, 128], [192, 0, 128], [64, 128, 128], [192, 128, 128], |
| | [0, 64, 0], [128, 64, 0], [0, 192, 0], [128, 192, 0], |
| | [0, 64, 128]]) |
| |
|
| | def __init__(self, |
| | ann_file, |
| | img_suffix='.jpg', |
| | seg_map_suffix='.png', |
| | **kwargs) -> None: |
| | super().__init__( |
| | img_suffix=img_suffix, |
| | seg_map_suffix=seg_map_suffix, |
| | ann_file=ann_file, |
| | **kwargs) |
| | assert fileio.exists(self.data_prefix['img_path'], |
| | self.backend_args) and osp.isfile(self.ann_file) |
| |
|
