bev-project/mmdet3d/ops/pointnet_modules/__init__.py

21 lines
475 B
Python
Raw Permalink Normal View History

2022-06-03 12:21:18 +08:00
from .builder import build_sa_module
from .paconv_sa_module import (
PAConvCUDASAModule,
PAConvCUDASAModuleMSG,
PAConvSAModule,
PAConvSAModuleMSG,
)
from .point_fp_module import PointFPModule
from .point_sa_module import PointSAModule, PointSAModuleMSG
__all__ = [
"build_sa_module",
"PointSAModuleMSG",
"PointSAModule",
"PointFPModule",
"PAConvSAModule",
"PAConvSAModuleMSG",
"PAConvCUDASAModule",
"PAConvCUDASAModuleMSG",
]