Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
mmdetection3d
Commits
1f9eb6c8
Commit
1f9eb6c8
authored
Nov 15, 2022
by
Xiangxu-0103
Committed by
ZwwWayne
Dec 03, 2022
Browse files
[Enhance] Refactor the import of `mmdet` (#1947)
* fix config * refactor import of `mmdet` * fix lint * fix lint
parent
c5bc9ac9
Changes
66
Hide whitespace changes
Inline
Side-by-side
Showing
6 changed files
with
10 additions
and
10 deletions
+10
-10
mmdet3d/structures/ops/iou3d_calculator.py
mmdet3d/structures/ops/iou3d_calculator.py
+1
-1
mmdet3d/utils/collect_env.py
mmdet3d/utils/collect_env.py
+1
-1
mmdet3d/utils/typing.py
mmdet3d/utils/typing.py
+1
-1
mmdet3d/visualization/local_visualizer.py
mmdet3d/visualization/local_visualizer.py
+4
-4
setup.cfg
setup.cfg
+2
-2
tools/dataset_converters/create_gt_database.py
tools/dataset_converters/create_gt_database.py
+1
-1
No files found.
mmdet3d/structures/ops/iou3d_calculator.py
View file @
1f9eb6c8
# Copyright (c) OpenMMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
import
torch
import
torch
from
mmdet.structures.bbox
import
bbox_overlaps
from
mmdet3d.registry
import
TASK_UTILS
from
mmdet3d.registry
import
TASK_UTILS
from
mmdet3d.structures.bbox_3d
import
get_box_type
from
mmdet3d.structures.bbox_3d
import
get_box_type
from
mmdet.structures.bbox
import
bbox_overlaps
@
TASK_UTILS
.
register_module
()
@
TASK_UTILS
.
register_module
()
...
...
mmdet3d/utils/collect_env.py
View file @
1f9eb6c8
# Copyright (c) OpenMMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
import
mmdet
from
mmengine.utils
import
get_git_hash
from
mmengine.utils
import
get_git_hash
from
mmengine.utils.dl_utils
import
collect_env
as
collect_base_env
from
mmengine.utils.dl_utils
import
collect_env
as
collect_base_env
import
mmdet
import
mmdet3d
import
mmdet3d
...
...
mmdet3d/utils/typing.py
View file @
1f9eb6c8
...
@@ -2,11 +2,11 @@
...
@@ -2,11 +2,11 @@
"""Collecting some commonly used type hint in MMDetection3D."""
"""Collecting some commonly used type hint in MMDetection3D."""
from
typing
import
List
,
Optional
,
Union
from
typing
import
List
,
Optional
,
Union
from
mmdet.models.task_modules.samplers
import
SamplingResult
from
mmengine.config
import
ConfigDict
from
mmengine.config
import
ConfigDict
from
mmengine.structures
import
InstanceData
from
mmengine.structures
import
InstanceData
from
mmdet3d.structures.det3d_data_sample
import
Det3DDataSample
from
mmdet3d.structures.det3d_data_sample
import
Det3DDataSample
from
mmdet.models.task_modules.samplers
import
SamplingResult
# Type hint of config data
# Type hint of config data
ConfigType
=
Union
[
ConfigDict
,
dict
]
ConfigType
=
Union
[
ConfigDict
,
dict
]
...
...
mmdet3d/visualization/local_visualizer.py
View file @
1f9eb6c8
...
@@ -8,6 +8,7 @@ import numpy as np
...
@@ -8,6 +8,7 @@ import numpy as np
from
matplotlib.collections
import
PatchCollection
from
matplotlib.collections
import
PatchCollection
from
matplotlib.patches
import
PathPatch
from
matplotlib.patches
import
PathPatch
from
matplotlib.path
import
Path
from
matplotlib.path
import
Path
from
mmdet.visualization
import
DetLocalVisualizer
from
mmengine.dist
import
master_only
from
mmengine.dist
import
master_only
from
mmengine.structures
import
InstanceData
from
mmengine.structures
import
InstanceData
from
mmengine.visualization.utils
import
check_type
,
tensor2ndarray
from
mmengine.visualization.utils
import
check_type
,
tensor2ndarray
...
@@ -19,7 +20,6 @@ from mmdet3d.structures import (BaseInstance3DBoxes, CameraInstance3DBoxes,
...
@@ -19,7 +20,6 @@ from mmdet3d.structures import (BaseInstance3DBoxes, CameraInstance3DBoxes,
Det3DDataSample
,
LiDARInstance3DBoxes
,
Det3DDataSample
,
LiDARInstance3DBoxes
,
PointData
,
points_cam2img
)
PointData
,
points_cam2img
)
from
mmdet3d.structures.bbox_3d.box_3d_mode
import
Box3DMode
from
mmdet3d.structures.bbox_3d.box_3d_mode
import
Box3DMode
from
mmdet.visualization
import
DetLocalVisualizer
from
.vis_utils
import
(
proj_camera_bbox3d_to_img
,
proj_depth_bbox3d_to_img
,
from
.vis_utils
import
(
proj_camera_bbox3d_to_img
,
proj_depth_bbox3d_to_img
,
proj_lidar_bbox3d_to_img
,
to_depth_mode
)
proj_lidar_bbox3d_to_img
,
to_depth_mode
)
...
@@ -708,9 +708,9 @@ class Det3DLocalVisualizer(DetLocalVisualizer):
...
@@ -708,9 +708,9 @@ class Det3DLocalVisualizer(DetLocalVisualizer):
and masks. Defaults to 0.3.
and masks. Defaults to 0.3.
step (int): Global step value to record. Defaults to 0.
step (int): Global step value to record. Defaults to 0.
"""
"""
classes
=
self
.
dataset_meta
.
get
(
'
classes
'
,
None
)
classes
=
self
.
dataset_meta
.
get
(
'
CLASSES
'
,
None
)
# For object detection datasets, no
palette
is saved
# For object detection datasets, no
PALETTE
is saved
palette
=
self
.
dataset_meta
.
get
(
'
palette
'
,
None
)
palette
=
self
.
dataset_meta
.
get
(
'
PALETTE
'
,
None
)
ignore_index
=
self
.
dataset_meta
.
get
(
'ignore_index'
,
None
)
ignore_index
=
self
.
dataset_meta
.
get
(
'ignore_index'
,
None
)
gt_data_3d
=
None
gt_data_3d
=
None
...
...
setup.cfg
View file @
1f9eb6c8
...
@@ -7,8 +7,8 @@ SPLIT_BEFORE_EXPRESSION_AFTER_OPENING_PAREN = true
...
@@ -7,8 +7,8 @@ SPLIT_BEFORE_EXPRESSION_AFTER_OPENING_PAREN = true
line_length = 79
line_length = 79
multi_line_output = 0
multi_line_output = 0
extra_standard_library = setuptools
extra_standard_library = setuptools
known_first_party =
mmdet,mmseg,
mmdet3d
known_first_party = mmdet3d
known_third_party = cv2,imageio,indoor3d_util,load_scannet_data,lyft_dataset_sdk,m2r,matplotlib,mmcv,nuimages,numba,numpy,nuscenes,pandas,plyfile,pycocotools,pyquaternion,pytest,pytorch_sphinx_theme,recommonmark,requests,scannet_utils,scipy,seaborn,shapely,skimage,sphinx,tensorflow,terminaltables,torch,trimesh,ts,waymo_open_dataset
known_third_party = cv2,imageio,indoor3d_util,load_scannet_data,lyft_dataset_sdk,m2r,matplotlib,mmcv,
mmdet,mmengine,
nuimages,numba,numpy,nuscenes,pandas,plyfile,pycocotools,pyquaternion,pytest,pytorch_sphinx_theme,recommonmark,requests,scannet_utils,scipy,seaborn,shapely,skimage,sphinx,tensorflow,terminaltables,torch,trimesh,ts,waymo_open_dataset
no_lines_before = STDLIB,LOCALFOLDER
no_lines_before = STDLIB,LOCALFOLDER
default_section = THIRDPARTY
default_section = THIRDPARTY
...
...
tools/dataset_converters/create_gt_database.py
View file @
1f9eb6c8
...
@@ -6,13 +6,13 @@ import mmcv
...
@@ -6,13 +6,13 @@ import mmcv
import
mmengine
import
mmengine
import
numpy
as
np
import
numpy
as
np
from
mmcv.ops
import
roi_align
from
mmcv.ops
import
roi_align
from
mmdet.evaluation
import
bbox_overlaps
from
mmengine
import
track_iter_progress
from
mmengine
import
track_iter_progress
from
pycocotools
import
mask
as
maskUtils
from
pycocotools
import
mask
as
maskUtils
from
pycocotools.coco
import
COCO
from
pycocotools.coco
import
COCO
from
mmdet3d.datasets
import
build_dataset
from
mmdet3d.datasets
import
build_dataset
from
mmdet3d.structures.ops
import
box_np_ops
as
box_np_ops
from
mmdet3d.structures.ops
import
box_np_ops
as
box_np_ops
from
mmdet.evaluation
import
bbox_overlaps
def
_poly2mask
(
mask_ann
,
img_h
,
img_w
):
def
_poly2mask
(
mask_ann
,
img_h
,
img_w
):
...
...
Prev
1
2
3
4
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment