Update app.py

app.py

@@ -149,8 +149,8 @@ def remove_bg(fl, count, mh, ms, md, lm):
     global fl_
     fr = cv2.imread(fl).astype(np.uint8)
     
-    b = 3
-    element = cv2.getStructuringElement(cv2.MORPH_RECT, (2 * b + 1, 2 * b + 1), (b, b))
+    #b = 3
+    #element = cv2.getStructuringElement(cv2.MORPH_RECT, (2 * b + 1, 2 * b + 1), (b, b))
     
     n = int((fr.shape[0]*fr.shape[1]) / (256*256))
     fr_bg = cv2.medianBlur(fr, 255)
@@ -162,19 +162,21 @@ def remove_bg(fl, count, mh, ms, md, lm):
     hsv = cv2.cvtColor(fr_diff, cv2.COLOR_BGR2HSV) # range: 180, 255, 255
     
     fr_diff = cv2.cvtColor(fr_diff, cv2.COLOR_BGR2GRAY)
+
+    ret,fr_diff = cv2.threshold(fr_diff,0,255,cv2.THRESH_BINARY+cv2.THRESH_OTSU)
     
     #md = 12
-    if lm == "median":
-        mh = np.median(hsv[:,:,0])
-        ms = np.median(hsv[:,:,1])
-        md = np.median(hsv[:,:,2])
-    elif lm == "average":
-        mh = np.average(hsv[:,:,0])
-        ms = np.average(hsv[:,:,1])
-        md = np.average(hsv[:,:,2])
-    bg = cv2.inRange(hsv, np.array([0,0,0]), np.array([mh,ms,md]))
-    fr_diff[bg>0] = 0
-    fr_diff[bg==0] = 255
+    #if lm == "median":
+    #    mh = np.median(hsv[:,:,0])
+    #    ms = np.median(hsv[:,:,1])
+    #    md = np.median(hsv[:,:,2])
+    #elif lm == "average":
+    #    mh = np.average(hsv[:,:,0])
+    #    ms = np.average(hsv[:,:,1])
+    #    md = np.average(hsv[:,:,2])
+    #bg = cv2.inRange(hsv, np.array([0,0,0]), np.array([mh,ms,md]))
+    #fr_diff[bg>0] = 0
+    #fr_diff[bg==0] = 255
     
     cv2.rectangle(fr_diff,(0,0),(fr_diff.shape[1]-1,fr_diff.shape[0]-1),(255,255,255),1)
     mask = cv2.floodFill(fr_diff, None, (0, 0), 255, 0, 0, (4 | cv2.FLOODFILL_FIXED_RANGE))[2] #(4 | cv.FLOODFILL_FIXED_RANGE | cv.FLOODFILL_MASK_ONLY | 255 << 8)