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
7a43a52a
Commit
7a43a52a
authored
May 13, 2020
by
liyinhao
Browse files
change names
parent
f7cf2bb0
Changes
5
Hide whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
9 additions
and
5 deletions
+9
-5
mmdet3d/core/evaluation/__init__.py
mmdet3d/core/evaluation/__init__.py
+2
-1
mmdet3d/core/evaluation/indoor_utils/__init__.py
mmdet3d/core/evaluation/indoor_utils/__init__.py
+3
-0
mmdet3d/core/evaluation/indoor_utils/eval.py
mmdet3d/core/evaluation/indoor_utils/eval.py
+1
-1
mmdet3d/core/evaluation/scannet_utils/__init__.py
mmdet3d/core/evaluation/scannet_utils/__init__.py
+0
-0
mmdet3d/datasets/scannet_dataset.py
mmdet3d/datasets/scannet_dataset.py
+3
-3
No files found.
mmdet3d/core/evaluation/__init__.py
View file @
7a43a52a
from
.class_names
import
dataset_aliases
,
get_classes
,
kitti_classes
from
.class_names
import
dataset_aliases
,
get_classes
,
kitti_classes
from
.indoor_utils
import
indoor_eval
from
.kitti_utils
import
kitti_eval
,
kitti_eval_coco_style
from
.kitti_utils
import
kitti_eval
,
kitti_eval_coco_style
__all__
=
[
__all__
=
[
'dataset_aliases'
,
'get_classes'
,
'kitti_classes'
,
'kitti_eval_coco_style'
,
'dataset_aliases'
,
'get_classes'
,
'kitti_classes'
,
'kitti_eval_coco_style'
,
'kitti_eval'
'kitti_eval'
,
'indoor_eval'
]
]
mmdet3d/core/evaluation/indoor_utils/__init__.py
0 → 100644
View file @
7a43a52a
from
.eval
import
indoor_eval
__all__
=
[
'indoor_eval'
]
mmdet3d/core/evaluation/
scannet
_utils/eval.py
→
mmdet3d/core/evaluation/
indoor
_utils/eval.py
View file @
7a43a52a
...
@@ -365,7 +365,7 @@ def boxes3d_depth_to_lidar(boxes3d, mid_to_bottom=True):
...
@@ -365,7 +365,7 @@ def boxes3d_depth_to_lidar(boxes3d, mid_to_bottom=True):
return
boxes3d_lidar
return
boxes3d_lidar
def
scannet
_eval
(
gt_annos
,
dt_annos
,
metric
,
class2type
):
def
indoor
_eval
(
gt_annos
,
dt_annos
,
metric
,
class2type
):
"""Scannet Evaluation.
"""Scannet Evaluation.
Evaluate the result of the detection.
Evaluate the result of the detection.
...
...
mmdet3d/core/evaluation/scannet_utils/__init__.py
deleted
100644 → 0
View file @
f7cf2bb0
mmdet3d/datasets/scannet_dataset.py
View file @
7a43a52a
...
@@ -211,13 +211,13 @@ class ScannetDataset(torch_data.Dataset):
...
@@ -211,13 +211,13 @@ class ScannetDataset(torch_data.Dataset):
def
evaluate
(
self
,
results
,
metric
=
None
,
logger
=
None
,
pklfile_prefix
=
None
):
def
evaluate
(
self
,
results
,
metric
=
None
,
logger
=
None
,
pklfile_prefix
=
None
):
results
=
self
.
_format_results
(
results
)
results
=
self
.
_format_results
(
results
)
from
mmdet3d.core.evaluation
.scannet_utils.eval
import
scannet
_eval
from
mmdet3d.core.evaluation
import
indoor
_eval
assert
(
'AP_IOU_THRESHHOLDS'
in
metric
)
assert
(
'AP_IOU_THRESHHOLDS'
in
metric
)
gt_annos
=
[
gt_annos
=
[
copy
.
deepcopy
(
info
[
'annos'
])
for
info
in
self
.
scannet_infos
copy
.
deepcopy
(
info
[
'annos'
])
for
info
in
self
.
scannet_infos
]
]
ap_result_str
,
ap_dict
=
scannet
_eval
(
gt_annos
,
results
,
metric
,
ap_result_str
,
ap_dict
=
indoor
_eval
(
gt_annos
,
results
,
metric
,
self
.
class2type
)
self
.
class2type
)
return
ap_dict
return
ap_dict
def
__len__
(
self
):
def
__len__
(
self
):
...
...
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