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
beae2426
Commit
beae2426
authored
Jul 06, 2020
by
zhangwenwei
Browse files
Merge branch 'fix_visualize' into 'master'
Fix visualize See merge request open-mmlab/mmdet.3d!124
parents
f09e8c62
bfb35d09
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
5 deletions
+6
-5
mmdet3d/core/visualizer/show_result.py
mmdet3d/core/visualizer/show_result.py
+6
-5
No files found.
mmdet3d/core/visualizer/show_result.py
View file @
beae2426
...
@@ -78,16 +78,17 @@ def show_result(points, gt_bboxes, pred_bboxes, out_dir, filename):
...
@@ -78,16 +78,17 @@ def show_result(points, gt_bboxes, pred_bboxes, out_dir, filename):
out_dir (str): Path of output directory
out_dir (str): Path of output directory
filename (str): Filename of the current frame.
filename (str): Filename of the current frame.
"""
"""
mmcv
.
mkdir_or_exist
(
out_dir
)
result_path
=
osp
.
join
(
out_dir
,
filename
)
mmcv
.
mkdir_or_exist
(
result_path
)
if
gt_bboxes
is
not
None
:
if
gt_bboxes
is
not
None
:
gt_bboxes
[:,
6
]
*=
-
1
gt_bboxes
[:,
6
]
*=
-
1
_write_oriented_bbox
(
gt_bboxes
,
osp
.
join
(
out_dir
,
_write_oriented_bbox
(
gt_bboxes
,
f
'
{
filename
}
_gt.ply'
))
osp
.
join
(
result_path
,
f
'
{
filename
}
_gt.ply'
))
if
points
is
not
None
:
if
points
is
not
None
:
_write_ply
(
points
,
osp
.
join
(
out_dir
,
f
'
{
filename
}
_points.obj'
))
_write_ply
(
points
,
osp
.
join
(
result_path
,
f
'
{
filename
}
_points.obj'
))
if
pred_bboxes
is
not
None
:
if
pred_bboxes
is
not
None
:
pred_bboxes
[:,
6
]
*=
-
1
pred_bboxes
[:,
6
]
*=
-
1
_write_oriented_bbox
(
pred_bboxes
,
_write_oriented_bbox
(
pred_bboxes
,
osp
.
join
(
out_dir
,
f
'
{
filename
}
_pred.ply'
))
osp
.
join
(
result_path
,
f
'
{
filename
}
_pred.ply'
))
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