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

fix import error

parent 5c5e43b3
...@@ -42,13 +42,25 @@ extensions = [ ...@@ -42,13 +42,25 @@ extensions = [
] ]
autodoc_mock_imports = [ autodoc_mock_imports = [
'cv2', 'matplotlib', 'nuscenes', 'PIL', 'pycocotools', 'pyquaternion', 'cv2',
'terminaltables', 'mmcv', 'mmdet', 'mmdet3d.version', 'matplotlib',
'mmdet3d.ops.ball_query', 'mmdet3d.ops.furthest_point_sample', 'nuscenes',
'mmdet3d.ops.gather_points', 'mmdet3d.ops.group_points', 'PIL',
'mmdet3d.ops.interpolate', 'mmdet3d.ops.roiaware_pool3d', 'pycocotools',
'mmdet3d.ops.spconv', 'mmdet3d.ops.voxel.voxel_layer', 'mmdet3d.ops.iou3d', 'pyquaternion',
'mmdet3d.ops.utils' '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 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) sigmoid_focal_loss)
from .ball_query import ball_query from .ball_query import ball_query
...@@ -15,7 +16,6 @@ from .roiaware_pool3d import (RoIAwarePool3d, points_in_boxes_batch, ...@@ -15,7 +16,6 @@ from .roiaware_pool3d import (RoIAwarePool3d, points_in_boxes_batch,
points_in_boxes_cpu, points_in_boxes_gpu) points_in_boxes_cpu, points_in_boxes_gpu)
from .sparse_block import (SparseBasicBlock, SparseBottleneck, from .sparse_block import (SparseBasicBlock, SparseBottleneck,
make_sparse_convmodule) make_sparse_convmodule)
from .utils import get_compiler_version, get_compiling_cuda_version
from .voxel import DynamicScatter, Voxelization, dynamic_scatter, voxelization from .voxel import DynamicScatter, Voxelization, dynamic_scatter, voxelization
__all__ = [ __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