95 lines
2.3 KiB
Python
95 lines
2.3 KiB
Python
from mmcv.ops import (
|
|
RoIAlign,
|
|
SigmoidFocalLoss,
|
|
get_compiler_version,
|
|
get_compiling_cuda_version,
|
|
nms,
|
|
roi_align,
|
|
sigmoid_focal_loss,
|
|
)
|
|
|
|
from .ball_query import ball_query
|
|
#from .feature_decorator import feature_decorator
|
|
from .furthest_point_sample import (
|
|
Points_Sampler,
|
|
furthest_point_sample,
|
|
furthest_point_sample_with_dist,
|
|
)
|
|
from .gather_points import gather_points
|
|
from .group_points import GroupAll, QueryAndGroup, group_points, grouping_operation
|
|
from .interpolate import three_interpolate, three_nn
|
|
from .knn import knn
|
|
from .norm import NaiveSyncBatchNorm1d, NaiveSyncBatchNorm2d
|
|
from .paconv import PAConv, PAConvCUDA, assign_score_withk
|
|
from .bev_pool import *
|
|
from .pointnet_modules import (
|
|
PAConvCUDASAModule,
|
|
PAConvCUDASAModuleMSG,
|
|
PAConvSAModule,
|
|
PAConvSAModuleMSG,
|
|
PointFPModule,
|
|
PointSAModule,
|
|
PointSAModuleMSG,
|
|
build_sa_module,
|
|
)
|
|
from .roiaware_pool3d import (
|
|
RoIAwarePool3d,
|
|
points_in_boxes_batch,
|
|
points_in_boxes_cpu,
|
|
points_in_boxes_gpu,
|
|
)
|
|
from .sparse_block import SparseBasicBlock, SparseBottleneck, make_sparse_convmodule
|
|
from .voxel import DynamicScatter, Voxelization, dynamic_scatter, voxelization
|
|
|
|
__all__ = [
|
|
"nms",
|
|
"soft_nms",
|
|
"RoIAlign",
|
|
"roi_align",
|
|
"get_compiler_version",
|
|
"get_compiling_cuda_version",
|
|
"NaiveSyncBatchNorm1d",
|
|
"NaiveSyncBatchNorm2d",
|
|
"batched_nms",
|
|
"Voxelization",
|
|
"voxelization",
|
|
"dynamic_scatter",
|
|
"DynamicScatter",
|
|
"sigmoid_focal_loss",
|
|
"SigmoidFocalLoss",
|
|
"SparseBasicBlock",
|
|
"SparseBottleneck",
|
|
"RoIAwarePool3d",
|
|
"points_in_boxes_gpu",
|
|
"points_in_boxes_cpu",
|
|
"make_sparse_convmodule",
|
|
"ball_query",
|
|
"knn",
|
|
"furthest_point_sample",
|
|
"furthest_point_sample_with_dist",
|
|
"three_interpolate",
|
|
"three_nn",
|
|
"gather_points",
|
|
"grouping_operation",
|
|
"group_points",
|
|
"GroupAll",
|
|
"QueryAndGroup",
|
|
"PointSAModule",
|
|
"PointSAModuleMSG",
|
|
"PointFPModule",
|
|
"points_in_boxes_batch",
|
|
"get_compiler_version",
|
|
"assign_score_withk",
|
|
"get_compiling_cuda_version",
|
|
"Points_Sampler",
|
|
"build_sa_module",
|
|
"PAConv",
|
|
"PAConvCUDA",
|
|
"PAConvSAModuleMSG",
|
|
"PAConvSAModule",
|
|
"PAConvCUDASAModule",
|
|
"PAConvCUDASAModuleMSG",
|
|
"bev_pool",
|
|
"feature_decorator",
|
|
]
|