mmdet/models/task_modules/assigners/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .approx_max_iou_assigner import ApproxMaxIoUAssigner
from .assign_result import AssignResult
from .atss_assigner import ATSSAssigner
from .base_assigner import BaseAssigner
from .center_region_assigner import CenterRegionAssigner
from .dynamic_soft_label_assigner import DynamicSoftLabelAssigner
from .grid_assigner import GridAssigner
from .hungarian_assigner import HungarianAssigner
from .iou2d_calculator import BboxOverlaps2D, BboxOverlaps2D_GLIP
from .match_cost import (BBoxL1Cost, BinaryFocalLossCost, ClassificationCost,
                         CrossEntropyLossCost, DiceCost, FocalLossCost,
                         IoUCost)
from .max_iou_assigner import MaxIoUAssigner
from .multi_instance_assigner import MultiInstanceAssigner
from .point_assigner import PointAssigner
from .region_assigner import RegionAssigner
from .sim_ota_assigner import SimOTAAssigner
from .task_aligned_assigner import TaskAlignedAssigner
from .topk_hungarian_assigner import TopkHungarianAssigner
from .uniform_assigner import UniformAssigner

__all__ = [
    'BaseAssigner', 'BinaryFocalLossCost', 'MaxIoUAssigner',
    'ApproxMaxIoUAssigner', 'AssignResult', 'PointAssigner', 'ATSSAssigner',
    'CenterRegionAssigner', 'GridAssigner', 'HungarianAssigner',
    'RegionAssigner', 'UniformAssigner', 'SimOTAAssigner',
    'TaskAlignedAssigner', 'TopkHungarianAssigner', 'BBoxL1Cost',
    'ClassificationCost', 'CrossEntropyLossCost', 'DiceCost', 'FocalLossCost',
    'IoUCost', 'BboxOverlaps2D', 'DynamicSoftLabelAssigner',
    'MultiInstanceAssigner', 'BboxOverlaps2D_GLIP'
]