Upload utils.py

model/utils.py

@@ -155,28 +155,27 @@ def plot_bev_detections(
     # 5) Draw each box
     seen_labels = set()
     for box, score, label in zip(boxes, scores, labels):
-        if label != 1: 
-            x, y, z, dx, dy, dz, yaw, *_ = box.cpu().numpy()
-            cls_idx = int(label)
-            cls_name = class_names[cls_idx]
-            color = colors[cls_idx]
-    
-            # example of stretching length for 'car' if you still want it
-            if cls_name.lower() == 'car':
-                dx *= 1.2
-    
-            rect = patches.Rectangle(
-                (x - dx/2, y - dy/2),
-                dx, dy,
-                angle=np.degrees(yaw),
-                linewidth=1.5,
-                edgecolor=color,
-                facecolor='none'
-            )
-            ax.add_patch(rect)
-    
-            # remember we saw this label so we can add it to legend once
-            seen_labels.add(cls_idx)
+        x, y, z, dx, dy, dz, yaw, *_ = box.cpu().numpy()
+        cls_idx = int(label)
+        cls_name = class_names[cls_idx]
+        color = colors[cls_idx]
+
+        # example of stretching length for 'car' if you still want it
+        if cls_name.lower() == 'car':
+            dx *= 1.2
+
+        rect = patches.Rectangle(
+            (x - dx/2, y - dy/2),
+            dx, dy,
+            angle=np.degrees(yaw),
+            linewidth=1.5,
+            edgecolor=color,
+            facecolor='none'
+        )
+        ax.add_patch(rect)
+
+        # remember we saw this label so we can add it to legend once
+        seen_labels.add(cls_idx)
 
     # 6) Legend only for seen classes
     legend_handles = []