Update mesh.py

mesh.py

@@ -18,7 +18,7 @@ def depth_edges_mask(depth):
     # Compute the gradient magnitude.
     depth_grad = np.sqrt(depth_dx ** 2 + depth_dy ** 2)
     # Compute the edge mask.
-    mask = depth_grad > 2
+    mask = depth_grad > 0.05
     return mask
 
 
@@ -26,9 +26,9 @@ def predict_depth(model, image):
     depth = model.infer_pil(image)
     return depth
 
-def get_mesh(depth, image, keep_edges=False, skew=0.5):
+def get_mesh(depth, image, keep_edges=False):
     # limit the size of the input image
-    pts3d = depth_to_points(depth[None], skew=skew)
+    pts3d = depth_to_points(depth[None])
     pts3d = pts3d.reshape(-1, 3)
 
     # Create a trimesh mesh from the points