Update app.py

app.py

@@ -174,16 +174,18 @@ def remove_bg(fl, count):
     fr_diff = cv2.dilate(cv2.erode(fr_diff, element), element)
 
     if count % 2: # odd: is photo without the flash
-        fr_mask = np.bitwise_not(cv2.cvtColor(cv2.imread(fl_).astype(np.uint8), cv2.COLOR_BGR2GRAY))
-        fr_mask = np.bitwise_and(fr_diff, fr_mask).astype(np.uint8)
+        fr_mask = cv2.cvtColor(cv2.imread(fl_).astype(np.uint8), cv2.COLOR_BGR2GRAY)
+        fr_not = np.bitwise_not(fr_mask)
+        fr_shadow = np.bitwise_and(fr_diff, fr_not).astype(np.uint8)
+        fr_fg = np.bitwise_or(fr_diff, fr_mask).astype(np.uint8)
         cv2.imwrite(fl_, fr_mask)
 
-        m = cv2.inRange(fr, np.array([240,240,240]), np.array([255,255,255]))
-        fr[m>0] = (239,239,239)
-        m = cv2.inRange(fr, np.array([0,0,0]), np.array([15,15,15]))
-        fr[m>0] = (16,16,16)
-        fr[fr_mask==255] = (fr[fr_mask==255] / 17).astype(np.uint8)
-        fr[fr_diff==0] = (255,255,255)
+        #m = cv2.inRange(fr, np.array([240,240,240]), np.array([255,255,255]))
+        #fr[m>0] = (239,239,239)
+        #m = cv2.inRange(fr, np.array([0,0,0]), np.array([15,15,15]))
+        #fr[m>0] = (16,16,16)
+        fr[fr_shadow>0] = (fr[fr_shadow>0] / 17).astype(np.uint8)
+        fr[fr_fg==0] = (255,255,255)
         
         cv2.imwrite(fl, fr)
         return fl