annotator/entityseg/mask2former/__init__.py

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

# config
from .config import add_maskformer2_config

# models
from .maskformer_model import MaskFormer
from .cropformer_model import CropFormer
from .test_time_augmentation import SemanticSegmentorWithTTA