mask_former/modeling/__init__.py

# Copyright (c) Facebook, Inc. and its affiliates.
from .backbone.swin import D2SwinTransformer
from .backbone.vit import D2ViTTransformer
from .heads.mask_former_head import MaskFormerHead
from .heads.per_pixel_baseline import PerPixelBaselineHead, PerPixelBaselinePlusHead
from .heads.pixel_decoder import BasePixelDecoder
from .heads.big_pixel_decoder import BigPixelDecoder
from .heads.mega_big_pixel_decoder import  MegaBigPixelDecoder
from .heads.mask_former_head_baseline import MaskFormerBaselineHead