mmdet3d/models/task_modules/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from mmdet.models.task_modules import AssignResult, BaseAssigner

from .anchor import (ANCHOR_GENERATORS, PRIOR_GENERATORS,
                     AlignedAnchor3DRangeGenerator,
                     AlignedAnchor3DRangeGeneratorPerCls,
                     Anchor3DRangeGenerator, build_anchor_generator,
                     build_prior_generator)
from .assigners import Max3DIoUAssigner
from .coders import (AnchorFreeBBoxCoder, CenterPointBBoxCoder,
                     DeltaXYZWLHRBBoxCoder, FCOS3DBBoxCoder,
                     GroupFree3DBBoxCoder, MonoFlexCoder,
                     PartialBinBasedBBoxCoder, PGDBBoxCoder,
                     PointXYZWHLRBBoxCoder, SMOKECoder)
from .samplers import (BaseSampler, CombinedSampler,
                       InstanceBalancedPosSampler, IoUBalancedNegSampler,
                       IoUNegPiecewiseSampler, OHEMSampler, PseudoSampler,
                       RandomSampler, SamplingResult)
from .voxel import VoxelGenerator

__all__ = [
    'BaseAssigner', 'Max3DIoUAssigner', 'AssignResult', 'BaseSampler',
    'PseudoSampler', 'RandomSampler', 'InstanceBalancedPosSampler',
    'IoUBalancedNegSampler', 'CombinedSampler', 'OHEMSampler',
    'SamplingResult', 'IoUNegPiecewiseSampler', 'DeltaXYZWLHRBBoxCoder',
    'PartialBinBasedBBoxCoder', 'CenterPointBBoxCoder', 'AnchorFreeBBoxCoder',
    'GroupFree3DBBoxCoder', 'PointXYZWHLRBBoxCoder', 'FCOS3DBBoxCoder',
    'PGDBBoxCoder', 'SMOKECoder', 'MonoFlexCoder', 'VoxelGenerator',
    'AlignedAnchor3DRangeGenerator', 'Anchor3DRangeGenerator',
    'build_prior_generator', 'AlignedAnchor3DRangeGeneratorPerCls',
    'build_anchor_generator', 'ANCHOR_GENERATORS', 'PRIOR_GENERATORS'
]