Spaces:

hamacojr
/

SAM-CAT-Seg

Runtime error

seokju cho

initial commit

f8f62f3 about 1 year ago

No virus

605 Bytes

	# Copyright (c) Facebook, Inc. and its affiliates.
	from . import data # register all new datasets
	from . import modeling

	# config
	from .config import add_cat_seg_config

	# dataset loading
	from .data.dataset_mappers.detr_panoptic_dataset_mapper import DETRPanopticDatasetMapper
	from .data.dataset_mappers.mask_former_panoptic_dataset_mapper import (
	MaskFormerPanopticDatasetMapper,
	)
	from .data.dataset_mappers.mask_former_semantic_dataset_mapper import (
	MaskFormerSemanticDatasetMapper,
	)

	# models
	from .cat_seg_model import CATSeg
	from .test_time_augmentation import SemanticSegmentorWithTTA