kmax_deeplab/__init__.py

from . import data  # register all new datasets
from . import modeling

# config
from .config import add_kmax_deeplab_config

# dataset loading
from .data.dataset_mappers.coco_panoptic_kmaxdeeplab_dataset_mapper import COCOPanoptickMaXDeepLabDatasetMapper


# models
from .kmax_model import kMaXDeepLab

# evaluation
from .evaluation.instance_evaluation import InstanceSegEvaluator