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
b107238d
Commit
b107238d
authored
Jun 20, 2020
by
zhangwenwei
Browse files
Merge branch 'clean-iou' into 'master'
clean iou calculation See merge request open-mmlab/mmdet.3d!84
parents
b2c43ffd
8c4c9aee
Changes
22
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
10 deletions
+4
-10
mmdet3d/utils/logger.py
mmdet3d/utils/logger.py
+0
-9
tests/test_parta2_bbox_head.py
tests/test_parta2_bbox_head.py
+4
-1
No files found.
mmdet3d/utils/logger.py
deleted
100644 → 0
View file @
b2c43ffd
import
logging
from
mmcv.utils
import
get_logger
def
get_root_logger
(
log_file
=
None
,
log_level
=
logging
.
INFO
):
logger
=
get_logger
(
name
=
'mmdet3d'
,
log_file
=
log_file
,
log_level
=
log_level
)
return
logger
tests/test_parta2_bbox_head.py
View file @
b107238d
...
@@ -425,7 +425,10 @@ def test_multi_class_nms():
...
@@ -425,7 +425,10 @@ def test_multi_class_nms():
1.4715e+00
,
-
2.9639e+00
1.4715e+00
,
-
2.9639e+00
]]).
cuda
()
]]).
cuda
()
selected
=
self
.
multi_class_nms
(
box_probs
,
box_preds
,
0.1
,
0.001
)
input_meta
=
dict
(
box_type_3d
=
LiDARInstance3DBoxes
,
box_mode_3d
=
Box3DMode
.
LIDAR
)
selected
=
self
.
multi_class_nms
(
box_probs
,
box_preds
,
0.1
,
0.001
,
input_meta
)
expected_selected
=
torch
.
Tensor
([
0
,
1
,
4
,
8
]).
cuda
()
expected_selected
=
torch
.
Tensor
([
0
,
1
,
4
,
8
]).
cuda
()
assert
torch
.
all
(
selected
==
expected_selected
)
assert
torch
.
all
(
selected
==
expected_selected
)
...
...
Prev
1
2
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