Skip to content

Commit

Permalink
bugfixed: ignore empty boxes in visualization (open-mmlab#234)
Browse files Browse the repository at this point in the history
  • Loading branch information
sshaoshuai authored Aug 10, 2020
1 parent 262596c commit 9e460da
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tools/visual_utils/visualize_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -157,7 +157,7 @@ def draw_scenes(points, gt_boxes=None, ref_boxes=None, ref_scores=None, ref_labe
corners3d = boxes_to_corners_3d(gt_boxes)
fig = draw_corners3d(corners3d, fig=fig, color=(0, 0, 1), max_num=100)

if ref_boxes is not None:
if ref_boxes is not None and len(ref_boxes) > 0:
ref_corners3d = boxes_to_corners_3d(ref_boxes)
if ref_labels is None:
fig = draw_corners3d(ref_corners3d, fig=fig, color=(0, 1, 0), cls=ref_scores, max_num=100)
Expand Down

0 comments on commit 9e460da

Please sign in to comment.