tianma

Add task-metric eval path and Cascade Mask R-CNN + Swin-B for det/instance.

5c11913 3 months ago

493 Bytes

from .datasets import (
    COCOImageDataset,
    COCOWholeBodyImageDataset,
    ImageFolderDataset,
    build_test_transform,
    build_train_transform,
    collate_keep,
)
from .coco_eval import COCOEvalDataset, TASK_ANN_FILES, coco_eval_collate

__all__ = [
    "COCOImageDataset",
    "COCOWholeBodyImageDataset",
    "ImageFolderDataset",
    "COCOEvalDataset",
    "TASK_ANN_FILES",
    "build_test_transform",
    "build_train_transform",
    "collate_keep",
    "coco_eval_collate",
]

	from .datasets import (
	COCOImageDataset,
	COCOWholeBodyImageDataset,
	ImageFolderDataset,
	build_test_transform,
	build_train_transform,
	collate_keep,
	)
	from .coco_eval import COCOEvalDataset, TASK_ANN_FILES, coco_eval_collate

	__all__ = [
	"COCOImageDataset",
	"COCOWholeBodyImageDataset",
	"ImageFolderDataset",
	"COCOEvalDataset",
	"TASK_ANN_FILES",
	"build_test_transform",
	"build_train_transform",
	"collate_keep",
	"coco_eval_collate",
	]