mmdetection/mmdet/models/task_modules/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .assigners import *  # noqa: F401,F403
from .builder import (ANCHOR_GENERATORS, BBOX_ASSIGNERS, BBOX_CODERS,
                      BBOX_SAMPLERS, IOU_CALCULATORS, MATCH_COSTS,
                      PRIOR_GENERATORS, build_anchor_generator, build_assigner,
                      build_bbox_coder, build_iou_calculator, build_match_cost,
                      build_prior_generator, build_sampler)
from .coders import *  # noqa: F401,F403
from .prior_generators import *  # noqa: F401,F403
from .samplers import *  # noqa: F401,F403
from .tracking import *  # noqa: F401,F403

__all__ = [
    'ANCHOR_GENERATORS', 'PRIOR_GENERATORS', 'BBOX_ASSIGNERS', 'BBOX_SAMPLERS',
    'MATCH_COSTS', 'BBOX_CODERS', 'IOU_CALCULATORS', 'build_anchor_generator',
    'build_prior_generator', 'build_assigner', 'build_sampler',
    'build_iou_calculator', 'build_match_cost', 'build_bbox_coder'
]