Commit 6ad0fbaa authored by zhangwenwei's avatar zhangwenwei
Browse files

fix import error

parent 5c5e43b3
......@@ -42,13 +42,25 @@ extensions = [
]
autodoc_mock_imports = [
'cv2', 'matplotlib', 'nuscenes', 'PIL', 'pycocotools', 'pyquaternion',
'terminaltables', 'mmcv', 'mmdet', 'mmdet3d.version',
'mmdet3d.ops.ball_query', 'mmdet3d.ops.furthest_point_sample',
'mmdet3d.ops.gather_points', 'mmdet3d.ops.group_points',
'mmdet3d.ops.interpolate', 'mmdet3d.ops.roiaware_pool3d',
'mmdet3d.ops.spconv', 'mmdet3d.ops.voxel.voxel_layer', 'mmdet3d.ops.iou3d',
'mmdet3d.ops.utils'
'cv2',
'matplotlib',
'nuscenes',
'PIL',
'pycocotools',
'pyquaternion',
'terminaltables',
'mmcv',
'mmdet',
'mmdet3d.version',
'mmdet3d.ops.ball_query',
'mmdet3d.ops.furthest_point_sample',
'mmdet3d.ops.gather_points',
'mmdet3d.ops.group_points',
'mmdet3d.ops.interpolate',
'mmdet3d.ops.roiaware_pool3d',
'mmdet3d.ops.spconv',
'mmdet3d.ops.voxel.voxel_layer',
'mmdet3d.ops.iou3d',
]
autosectionlabel_prefix_document = True
......
from mmcv.ops import (RoIAlign, SigmoidFocalLoss, nms, roi_align,
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
......@@ -15,7 +16,6 @@ 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 .utils import get_compiler_version, get_compiling_cuda_version
from .voxel import DynamicScatter, Voxelization, dynamic_scatter, voxelization
__all__ = [
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment