Update model/utils.py

model/utils.py

@@ -12,8 +12,8 @@ def plot_bev_detections(
 ):
 
     class_names = [
-        'truck', 'construction_vehicle', 'bus', 'trailer', 'barrier', 'car', 
-        'motorcycle', 'bicycle', 'traffic_cone', 'pedestrian'
+        'truck', 'construction_vehicle', 'bus', 'trailer', 'barrier', 'pedestrian', 
+        'motorcycle', 'bicycle', 'traffic_cone', 'car'
     ]
 
     # 1) Create figure & axes
@@ -36,7 +36,8 @@ def plot_bev_detections(
     cmap = plt.get_cmap('tab10')  # up to 10 distinct colors
     num_classes = len(class_names)
     colors = {i: cmap(i % 10) for i in range(num_classes)}
-
+    colors[1] = 'white'
+    
     # 5) Draw each box
     seen_labels = set()
     for box, score, label in zip(boxes, scores, labels):
@@ -65,9 +66,10 @@ def plot_bev_detections(
     # 6) Legend only for seen classes
     legend_handles = []
     for cls_idx in sorted(seen_labels):
-        legend_handles.append(
-            patches.Patch(color=colors[cls_idx], label=class_names[cls_idx])
-        )
+        if cls_idx != 1:
+            legend_handles.append(
+                patches.Patch(color=colors[cls_idx], label=class_names[cls_idx])
+            )
     ax.legend(handles=legend_handles, loc='upper right')
 
     # 7) Axes limits and labels