Update app.py

app.py

@@ -24,18 +24,22 @@ def find_tip(points, convex_hull):
 
 
 
+
+
+
 def infer(image_in):
-    img = cv2.imread(image_in)
+    img = cv2.imread("arrows.png")
 
-    contours, _ = cv2.findContours(preprocess(img), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+    contours, hierarchy = cv2.findContours(preprocess(img), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)
 
     for cnt in contours:
-        approx = cv2.approxPolyDP(cnt, 0.02 * cv2.arcLength(cnt, True), True)
+        peri = cv2.arcLength(cnt, True)
+        approx = cv2.approxPolyDP(cnt, 0.025 * peri, True)
         hull = cv2.convexHull(approx, returnPoints=False)
         sides = len(hull)
 
         if 6 > sides > 3 and sides + 2 == len(approx):
-            arrow_tip = find_tip(approx[:, 0, :], hull.squeeze())
+            arrow_tip = find_tip(approx[:,0,:], hull.squeeze())
             if arrow_tip:
                 cv2.drawContours(img, [cnt], -1, (0, 255, 0), 3)
                 cv2.circle(img, arrow_tip, 3, (0, 0, 255), cv2.FILLED)