from .coco import CocoDataset
from .registry import DATASETS


@DATASETS.register_module
class YolkDataSet(CocoDataset):

    CLASSES = ('yolk')
from .coco import CocoDataset
from .registry import DATASETS


@DATASETS.register_module
class YolkDataSet(CocoDataset):

    CLASSES = ('yolk')