Update app.py

app.py

@@ -181,8 +181,8 @@ def remove_bg(fl, count, mh, ms, md, lm):
     # 255 << 8 tells to fill with the value 255)
     mask = mask[1:mask.shape[0]-1, 1:mask.shape[1]-1]
     fr_diff[mask>0] = 0
-    
-    fr_diff = cv2.dilate(cv2.erode(fr_diff, element), element)
+
+    #fr_diff = cv2.dilate(cv2.erode(fr_diff, element), element)
 
     if count % 2: # odd: is photo without the flash
         fr_mask = cv2.cvtColor(cv2.imread(fl_).astype(np.uint8), cv2.COLOR_BGR2GRAY)
@@ -205,6 +205,12 @@ def remove_bg(fl, count, mh, ms, md, lm):
         cv2.imwrite(fl_, fr_diff.astype(np.uint8))
         return fl_
 
+def denoise(fl):
+    fr = cv2.imread(fl).astype(np.uint8)
+    fr = cv2.fastNlMeansDenoisingColored(fr, None, 10,10,7,21)
+    cv2.imwrite(fl, fr)
+    return fl
+
 def sharpest(fl, i):
     break_vid = get_frames(fl, "vid_input_frame", "origin", i)
 
@@ -251,6 +257,8 @@ def loadf(f, r_bg, mh, ms, md, lm):
             ftype = fl.split('/')
             if ftype[len(ftype)-1].split('.')[1] == 'mp4':
                 fl = sharpest(fl, i)
+            else:
+                fl = denoise(fl)
 
             if r_bg == True:
                 fl = remove_bg(fl, i, mh, ms, md, lm)