"build/demos/CMakeFiles/demo_imi_flat.dir/build.make" did not exist on "395d2ce606314a6729939084e5f492f37cd2ff13"
__init__.py 716 Bytes
Newer Older
Shaoshuai Shi's avatar
Shaoshuai Shi committed
1
from .anchor_head_multi import AnchorHeadMulti
2
from .anchor_head_single import AnchorHeadSingle
Shaoshuai Shi's avatar
Shaoshuai Shi committed
3
from .anchor_head_template import AnchorHeadTemplate
4
from .point_head_box import PointHeadBox
Shaoshuai Shi's avatar
Shaoshuai Shi committed
5
6
from .point_head_simple import PointHeadSimple
from .point_intra_part_head import PointIntraPartOffsetHead
7
from .center_head import CenterHead
yukang's avatar
yukang committed
8
from .voxelnext_head import VoxelNeXtHead
9
10
11
12
13

__all__ = {
    'AnchorHeadTemplate': AnchorHeadTemplate,
    'AnchorHeadSingle': AnchorHeadSingle,
    'PointIntraPartOffsetHead': PointIntraPartOffsetHead,
Gus-Guo's avatar
Gus-Guo committed
14
    'PointHeadSimple': PointHeadSimple,
15
    'PointHeadBox': PointHeadBox,
Gus-Guo's avatar
Gus-Guo committed
16
    'AnchorHeadMulti': AnchorHeadMulti,
yukang's avatar
yukang committed
17
18
    'CenterHead': CenterHead,
    'VoxelNeXtHead': VoxelNeXtHead,
19
}