File size: 1,736 Bytes
c2ca15f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
# 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'
]
|