my_dataset

from .registry import DATASETS
from .voc import VOCDataset


@DATASETS.register_module
class MyDataset(VOCDataset):

    # CLASSES = ('aeroplane', 'bicycle', 'bird', 'boat', 'bottle', 'bus', 'car',
    #            'cat', 'chair', 'cow', 'diningtable', 'dog', 'horse',
    #            'motorbike', 'person', 'pottedplant', 'sheep', 'sofa', 'train',
    #            'tvmonitor')

    # CLASSES = ('face', 'l', 's')
    # CLASSES = ('bottle', 'tvmonitor', 'person', 'bicycle', 'diningtable', 'boat', 'sheep',
    #            'train', 'dog', 'aeroplane', 'chair', 'bird')
    # CLASSES = ('Other', 'Blue', 'Red', 'White', 'Yellow', 'Black')
    CLASSES = ('Cola', 'Sprite', 'Fanta', 'Mirinda', 'Ganten')