Update app.py
Browse files
app.py
CHANGED
@@ -39,7 +39,7 @@ def show_preds_image(image_path):
|
|
39 |
if (int(cls)<3) and (conf > 0.70):
|
40 |
|
41 |
points = np.int32([mask])
|
42 |
-
|
43 |
color_number = classes_ids.index(int(box.cls[0]))
|
44 |
color = colors[color_number]
|
45 |
|
@@ -88,7 +88,7 @@ def show_preds_webcam(pil_image):
|
|
88 |
if (int(cls)<3) and (conf > 0.70):
|
89 |
|
90 |
points = np.int32([mask])
|
91 |
-
|
92 |
color_number = classes_ids.index(int(box.cls[0]))
|
93 |
color = colors[color_number]
|
94 |
|
|
|
39 |
if (int(cls)<3) and (conf > 0.70):
|
40 |
|
41 |
points = np.int32([mask])
|
42 |
+
cv2.polylines(img, points, True, (255, 0, 0), 1)
|
43 |
color_number = classes_ids.index(int(box.cls[0]))
|
44 |
color = colors[color_number]
|
45 |
|
|
|
88 |
if (int(cls)<3) and (conf > 0.70):
|
89 |
|
90 |
points = np.int32([mask])
|
91 |
+
cv2.polylines(img, points, True, (255, 0, 0), 1)
|
92 |
color_number = classes_ids.index(int(box.cls[0]))
|
93 |
color = colors[color_number]
|
94 |
|