maskrcnn_benchmark/data/datasets/__init__.py

# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
from .coco import COCODataset
from .voc import PascalVOCDataset
from .concat_dataset import ConcatDataset
from .background import Background
from .tsv import TSVDataset, ODTSVDataset

from .modulated_coco import ModulatedDataset, CocoDetection, CocoGrounding
from .flickr import FlickrDataset
from .refexp import RefExpDataset
from .mixed import MixedDataset
from .gqa import GQADataset

from .coco_dt import CocoDetectionTSV
from .caption import CaptionTSV
from .lvis import LvisDetection
from .pseudo_data import PseudoData
from .phrasecut import PhrasecutDetection
try:
    from .omnilabel import OmniLabelDataset
except:
    pass
__all__ = [
    "COCODataset",
    "TSVDataset",
    "ODTSVDataset",
    "ConcatDataset",
    "PascalVOCDataset",
    "Background",
    "ModulatedDataset",
    "MixedDataset",
    "CocoDetection",
    "FlickrDataset",
    "RefExpDataset",
    "GQADataset",
    "CocoDetectionTSV",
    "CocoGrounding",
    "CaptionTSV",
    "LvisDetection",
    "PseudoData",
    "PhrasecutDetection",
    "OmniLabelDataset",
]