Spaces:

Huu076
/

Yolo

Runtime error

Huu076 commited on Jun 12

Commit

79b1612

•

1 Parent(s): 3e4259c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -39,7 +39,7 @@ def show_preds_image(image_path):
             if (int(cls)<3) and (conf > 0.70):
                 points = np.int32([mask])
-                # cv2.polylines(img, points, True, (255, 0, 0), 1)
                 color_number = classes_ids.index(int(box.cls[0]))
                 color = colors[color_number]
@@ -88,7 +88,7 @@ def show_preds_webcam(pil_image):
             if (int(cls)<3) and (conf > 0.70):
                 points = np.int32([mask])
-                # cv2.polylines(img, points, True, (255, 0, 0), 1)
                 color_number = classes_ids.index(int(box.cls[0]))
                 color = colors[color_number]

             if (int(cls)<3) and (conf > 0.70):
                 points = np.int32([mask])
+                cv2.polylines(img, points, True, (255, 0, 0), 1)
                 color_number = classes_ids.index(int(box.cls[0]))
                 color = colors[color_number]
             if (int(cls)<3) and (conf > 0.70):
                 points = np.int32([mask])
+                cv2.polylines(img, points, True, (255, 0, 0), 1)
                 color_number = classes_ids.index(int(box.cls[0]))
                 color = colors[color_number]