Table of Contents
Shortcuts

# Source code for mmseg.datasets.voc

    import os.path as osp

    from .builder import DATASETS
    from .custom import CustomDataset

    @DATASETS.register_module()
    class PascalVOCDataset(CustomDataset):
        """Pascal VOC dataset.

        Args:
            split (str): Split txt file for Pascal VOC.
        """

        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, split, **kwargs):
            super(PascalVOCDataset, self).__init__(
                img_suffix='.jpg', seg_map_suffix='.png', split=split, **kwargs)
            assert osp.exists(self.img_dir) and self.split is not None