Update app.py

app.py

@@ -165,7 +165,7 @@ def remove_bg(fl, s, l, v):
             blur_s[i,j] = (d/512).astype(np.uint8)
 
     #remove regions of low saturation and lightness (get scene without shadow)
-    m = cv2.inRange(cv2.cvtColor(frame_c, cv2.COLOR_RGB2HSV), np.array([0,0,0]), np.array([180,s,l]))
+    m = cv2.inRange(cv2.cvtColor(frame_c, cv2.COLOR_BGR2HSV), np.array([0,0,0]), np.array([180,s,l]))
     mask = cv2.inRange(blur_s, 0, v)
     masks = np.bitwise_and(m, mask)
     frame_[masks==0] = (0,0,0)
@@ -196,15 +196,24 @@ def remove_bg(fl, s, l, v):
     m_ = cv2.inRange(frame_, np.array([1,1,1]), np.array([127,127,127]))
     frame_[m_>0] = (127,127,127)
 
-    
+
     #shadow is black, bg is white, fg is gray
-    frame_ = 255 - frame_
-    m_ = cv2.inRange(frame_, np.array([255,255,255]), np.array([255,255,255]))
-    frame_[m_>0] = (127,127,127)
-    m_ = cv2.inRange(frame_, np.array([128,128,128]), np.array([128,128,128]))
-    frame_[m_>0] = (255,255,255)
+    frame_ = 255 - cv2.cvtColor(frame_, cv2.COLOR_BGR2GRAY)
+    m_ = cv2.inRange(frame_, 255, 255)
+    frame_[m_>0] = 127
+    m_ = cv2.inRange(frame_, 128, 128)
+    frame_[m_>0] = 255
 
-    cv2.imwrite(fl, frame_)
+
+    #apply mask to output
+    m = cv2.inRange(frame, np.array([240,240,240]), np.array([255,255,255]))
+    frame[m>0] = (239,239,239)
+    m = cv2.inRange(frame, np.array([0,0,0]), np.array([15,15,15]))
+    frame[m>0] = (16,16,16)
+    frame[frame_==0] = (frame / 17).astype(np.uint8)
+    frame[frame_==255] = (255,255,255)
+
+    cv2.imwrite(fl, frame)
     return fl
 
 def logscale(linear):