Spaces:

umichVision
/

virtex-redcaps

Runtime error

virtex-redcaps / virtex /data /__init__.py

Black + isort, remove unused virtx files.

8d0e872 over 2 years ago

No virus

629 Bytes

	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",
	]