Update app.py

app.py

@@ -205,15 +205,21 @@ def remove_bg(fl, count, mh, ms, md, lm):
         
         cv2.imwrite(fl, fr)
 
-        #b = 15
-        #element = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2 * b + 1, 2 * b + 1), (b, b))
-        #mask_e = cv2.erode(mask, element) * 255
-        #mask_d = cv2.dilate(mask, element) * 127
-        #mask_d[mask_e>0] = 255
-        #cv2.imwrite(f"{str(count)}_trimask.png", mask_d.astype(np.uint8))
-
-        #cutout(fl, f"{str(count)}_trimask.png",  f"{str(count)}_cutout.png")
-        
+        b = 15
+        d = 1
+        element = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2 * b + 1, 2 * b + 1), (b, b))
+        mask_e = cv2.erode(mask, element) * 255
+        element = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2 * d + 1, 2 * d + 1), (d, d))
+        mask_d = cv2.dilate(mask, element) * 127
+        mask_d[mask_e>0] = 255
+        cv2.imwrite(f"{str(count)}_trimask.png", mask_d.astype(np.uint8))
+        cutout(fl, f"{str(count)}_trimask.png",  f"{str(count)}_cutout.png")
+
+        a_map = cv2.imread(f"{str(count)}_cutout.png", cv2.IMREAD_UNCHANGED).astype(np.uint8)
+        a_map[:,:,3] = ((255 - a_map[:,:,3]) / 255).astype(np.float16)
+        fr[a_map[:,:,3]>0] = (fr * a_map[:,:,3] * 15 + 240).astype(np.uint8)
+
+        cv2.imwrite(fl, fr)
         return fl #f"{str(count)}_cutout.png"
     else: # even: with the flash
         fl_ = fl.split(".")[0] + "_.png"
@@ -222,7 +228,7 @@ def remove_bg(fl, count, mh, ms, md, lm):
 
 def denoise(fl):
     fr = cv2.imread(fl).astype(np.uint8)
-    fr = cv2.fastNlMeansDenoisingColored(fr, None, 10,10,7,21)
+    fr = cv2.fastNlMeansDenoisingColored(fr, None, 5,5,7,21)
     cv2.imwrite(fl, fr)
     return fl
 
@@ -252,7 +258,7 @@ def sharpest(fl, i):
         imgToDenoiseIndex = indx,
         temporalWindowSize = n,
         hColor = 5,
-        templateWindowSize = 21,
+        templateWindowSize = 7,
         searchWindowSize = 21)
 
     cv2.imwrite(fl, frame)