__init__.py 247 Bytes
Newer Older
1
2
from .spconv_backbone import VoxelBackBone8x
from .spconv_unet import UNetV2
Shaoshuai Shi's avatar
Shaoshuai Shi committed
3
from .pointnet2_backbone import PointNet2Backbone
4
5
6

__all__ = {
    'VoxelBackBone8x': VoxelBackBone8x,
Shaoshuai Shi's avatar
Shaoshuai Shi committed
7
8
    'UNetV2': UNetV2,
    'PointNet2Backbone': PointNet2Backbone
9
}