vicellst-official

Sleeping

polejowska commited on May 24, 2023

Commit

823e366

1 Parent(s): 434d15a

Update visualization.py

Files changed (1) hide show

visualization.py CHANGED Viewed

@@ -39,7 +39,7 @@ def visualize_prediction(
     for score, (xmin, ymin, xmax, ymax), label, color in zip(
         scores, boxes, labels, colors
     ):
-        counter =+ 1
         ax.add_patch(
             plt.Rectangle(
                 (xmin, ymin),

     for score, (xmin, ymin, xmax, ymax), label, color in zip(
         scores, boxes, labels, colors
     ):
+        counter += 1
         ax.add_patch(
             plt.Rectangle(
                 (xmin, ymin),