Update app.py

app.py

@@ -299,16 +299,10 @@ def get_mesh(image, depth, blur_data, loadall):
         clrs = np.concatenate((clrs, colors), axis=0)
         i = i+1
 
-    clrs = np.array([clrs])
-    verts = np.array([verts])
-    mask = cv2.inRange(verts, np.array([0.0,0.0,0.0]), np.array([0.0,0.0,0.0]))
-    clrs = clrs[mask==0]
-    verts = verts[mask==0]
-    print(clrs)
-    print(verts)
-    print(mask)
     #mesh = trimesh.Trimesh(vertices=verts, faces=triangles, vertex_colors=colors)
-    mesh.append(trimesh.PointCloud(verts[0].tolist(), colors=clrs[0].tolist()))
+    mesh.append(trimesh.PointCloud(verts, colors=clrs))
+    mask = (mesh[len(mesh)-1].base.vertices != [0.0,0.0,0.0]).any(axis=1)
+    mesh[len(mesh)-1].update_vertices(mask)
     #material = trimesh.visual.texture.SimpleMaterial(image=image)
     #texture = trimesh.visual.TextureVisuals(uv=uv, image=image, material=material)
     #mesh.visual = texture