Update app.py

app.py

@@ -418,15 +418,17 @@ def optimize(v, d):
 
             dcolor.append(bincount(frame))
             print(dcolor[k])
-            depth = cv2.imread(depths[k]).astype(np.uint8)
-            mask = cv2.inRange(frame, np.array([dcolor[k][0]-8, dcolor[k][1]-8, dcolor[k][2]-8]), np.array([dcolor[k][0]+8, dcolor[k][1]+8, dcolor[k][2]+8]))
-            depth[mask>0] = (255,255,255)
-            cv2.imwrite(depths[k], depth)
             
             mask = cv2.convertScaleAbs(cv2.Laplacian(cv2.cvtColor(frame, cv2.COLOR_RGB2GRAY), ddepth, ksize=kernel_size))
             mask[mask>0] = 255
             frame[mask==0] = (0, 0, 0)
             cv2.imwrite(frames[k], frame)
+
+            depth = cv2.imread(depths[k]).astype(np.uint8)
+            depth[mask==0] = (255,255,255)
+            mask = cv2.inRange(frame, np.array([dcolor[k][0]-8, dcolor[k][1]-8, dcolor[k][2]-8]), np.array([dcolor[k][0]+8, dcolor[k][1]+8, dcolor[k][2]+8]))
+            depth[mask>0] = (255,255,255)
+            cv2.imwrite(depths[k], depth)
             
     if d == False:
       return frames, "[" + str(dcolor[frame_selected])[1:-1] + ", 255]"