| |
| 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) |
|
|
