22 lines
802 B
Python
22 lines
802 B
Python
|
from .anchor_head_multi import AnchorHeadMulti
|
||
|
|
from .anchor_head_single import AnchorHeadSingle
|
||
|
|
from .anchor_head_template import AnchorHeadTemplate
|
||
|
|
from .point_head_box import PointHeadBox
|
||
|
|
from .point_head_simple import PointHeadSimple
|
||
|
|
from .point_intra_part_head import PointIntraPartOffsetHead
|
||
|
|
from .center_head import CenterHead
|
||
|
|
from .voxelnext_head import VoxelNeXtHead
|
||
|
|
from .transfusion_head import TransFusionHead
|
||
|
|
|
||
|
|
__all__ = {
|
||
|
|
'AnchorHeadTemplate': AnchorHeadTemplate,
|
||
|
|
'AnchorHeadSingle': AnchorHeadSingle,
|
||
|
|
'PointIntraPartOffsetHead': PointIntraPartOffsetHead,
|
||
|
|
'PointHeadSimple': PointHeadSimple,
|
||
|
|
'PointHeadBox': PointHeadBox,
|
||
|
|
'AnchorHeadMulti': AnchorHeadMulti,
|
||
|
|
'CenterHead': CenterHead,
|
||
|
|
'VoxelNeXtHead': VoxelNeXtHead,
|
||
|
|
'TransFusionHead': TransFusionHead,
|
||
|
|
}
|