Update app.py

app.py

@@ -484,14 +484,9 @@ def apply_mask(d, b):
     dmask = cv2.dilate(mask, dilation)
     dmask_b = cv2.GaussianBlur(dmask, (b,b), 0)
 
-    d["background"][mask==255] = (0,0,0,0)
-    mask = cv2.inRange(d["background"], np.array([0,0,0,0]), np.array([0,0,0,0]))
-
     for k, mk in enumerate(masks):
         if k != frame_selected and k < len(depths):
-            #masks[k] = cv2.imread(depths[k]).astype(np.uint8)
-            #masks[k][dmask<255] = (0,0,0)
-            #cv2.imwrite(depths[k], masks[k])
+            masks[k][dmask<255] = 0
             frame = cv2.imread(frames[k], cv2.IMREAD_UNCHANGED).astype(np.uint8)
             frame[:, :, 3] = dmask_b
             cv2.imwrite(frames[k], frame)
@@ -501,10 +496,10 @@ def apply_mask(d, b):
     cv2.imwrite(frames[frame_selected], frame)
     
     masks[frame_selected] = mask #d["background"]
-    #cv2.imwrite(depths[frame_selected], masks[frame_selected])
     return masks[frame_selected], depths, frames
 
 def draw_mask(l, t, v, d, evt: gr.EventData):
+    global depths
     global params
     global frame_selected
     global masks
@@ -521,7 +516,7 @@ def draw_mask(l, t, v, d, evt: gr.EventData):
 
     if len(edge) == 0 or params["fnum"] != frame_selected or params["l"] != l:
       if len(edge) > 0:
-          d["background"] = cv2.imread(masks[frame_selected]).astype(np.uint8)
+          d["background"] = cv2.imread(depths[frame_selected]).astype(np.uint8)
 
       bg = cv2.cvtColor(d["background"], cv2.COLOR_RGBA2GRAY)
 
@@ -643,7 +638,7 @@ async(c, o, b, p, d, n, m)=>{
             let mesh = q[i];
             mesh.dispose(false, true);
           }
-          BABYLON.Engine.LastCreatedScene.meshes[BABYLON.Engine.LastCreatedScene.meshes.length-1].material.needDepthPrePass = true;
+          //BABYLON.Engine.LastCreatedScene.meshes[BABYLON.Engine.LastCreatedScene.meshes.length-1].material.needDepthPrePass = true;
           BABYLON.Engine.LastCreatedScene.meshes[BABYLON.Engine.LastCreatedScene.meshes.length-1].alphaIndex = o_.length-j;
           BABYLON.Engine.LastCreatedScene.meshes[BABYLON.Engine.LastCreatedScene.meshes.length-1].scaling.z = -1;
           BABYLON.Engine.LastCreatedScene.meshes[BABYLON.Engine.LastCreatedScene.meshes.length-1].applyDisplacementMap(m[o_[j]].url, 0, 255, function(m){try{alert(BABYLON.Engine.Version);}catch(e){alert(e);}}, null, null, true, function(e){alert(e);});
@@ -676,11 +671,7 @@ async(c, o, b, p, d, n, m)=>{
           o_[j] = parseInt(o_[j]);
           alert(o_[j]);
           document.getElementById("iframe3D").contentDocument.getElementById("bgimage").value += p[o_[j]].image.url + ",";
-          if (o_[j] == n) {
-            document.getElementById("iframe3D").contentDocument.getElementById("bgdepth").value += d.background.url + ",";
-          } else {
-            document.getElementById("iframe3D").contentDocument.getElementById("bgdepth").value += m[o_[j]].url + ",";
-          }
+          document.getElementById("iframe3D").contentDocument.getElementById("bgdepth").value += m[o_[j]].url + ",";
         }
       }
       toggleDisplay("model");