Spaces:

yaghi27
/

ImageToBEV-lightweight

Runtime error

yaghi27 commited on Aug 6

Commit

3d641e6

1 Parent(s): 5d8ed5e

Update model/utils.py

Files changed (1) hide show

model/utils.py CHANGED Viewed

@@ -36,7 +36,6 @@ 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()

     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)}
     # 5) Draw each box
     seen_labels = set()