Add check for empty prediction labels (#1725)

open-mmlab · Aug 17, 2022 · d21fa99 · d21fa99
1 parent 903a9d2
commit d21fa99
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmdet3d/core/visualizer/show_result.py b/mmdet3d/core/visualizer/show_result.py
@@ -103,7 +103,7 @@ def show_result(points,
 
  vis = Visualizer(points)
  if pred_bboxes is not None:
- if pred_labels is None:
+ if pred_labels is None or pred_labels.numel() == 0:
  vis.add_bboxes(bbox3d=pred_bboxes)
  else:
  palette = np.random.randint(