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(