virtex/data/__init__.py

from .datasets.captioning import CaptioningDataset
from .datasets.classification import (
    TokenClassificationDataset,
    MultiLabelClassificationDataset,
)
from .datasets.masked_lm import MaskedLmDataset
from .datasets.downstream import (
    ImageNetDataset,
    INaturalist2018Dataset,
    VOC07ClassificationDataset,
    ImageDirectoryDataset,
)


__all__ = [
    "CaptioningDataset",
    "TokenClassificationDataset",
    "MultiLabelClassificationDataset",
    "MaskedLmDataset",
    "ImageDirectoryDataset",
    "ImageNetDataset",
    "INaturalist2018Dataset",
    "VOC07ClassificationDataset",
    "TarfileDataset",
]