Update app.py

app.py

@@ -9,7 +9,7 @@ import torch.nn.functional as F
 from torchvision import transforms
 from torchvision.transforms import Compose
 import trimesh
-from geometry import create_triangles
+#from geometry import create_triangles
 import tempfile
 from functools import partial
 import spaces
@@ -321,41 +321,30 @@ def get_mesh(image, depth, blur_data, loadall):
             #if (clrs[j] == pcolors[i]).all():
                 #cverts[i].append(verts[j])
                 #j=j+1
-    #mesh.append(trimesh.PointCloud(verts, colors=[255,0,255,255]))
+    #mesh.append(trimesh.PointCloud(verts, colors=clrs))
     #print(mesh[len(mesh)-1])
     #mesh[len(mesh)-1].merge_vertices()
         #i=i+1
     mesh_n.append(str(fnum))
 
-    triangles = create_triangles(rgba.shape[0], rgba.shape[1])
-    rgba_pil = Image.fromarray(rgba.astype(np.uint8))
-    material = trimesh.visual.texture.SimpleMaterial(image=rgba_pil) #, ambient=[0,0,0,0], diffuse=[0,0,0,0], specular=[0,0,0,0], glossiness=1.0)
-    visuals = trimesh.visual.TextureVisuals(uv=uvs, image=rgba_pil, material=material)
-    mesh.append(trimesh.Trimesh(vertices=verts, faces=triangles, vertex_normals=normals, visual=visuals, validate=True, process=False))
+    #triangles = create_triangles(rgba.shape[0], rgba.shape[1])
+    #rgba_pil = Image.fromarray(rgba.astype(np.uint8))
+    #material = trimesh.visual.texture.SimpleMaterial(image=rgba_pil) #, ambient=[0,0,0,0], diffuse=[0,0,0,0], specular=[0,0,0,0], glossiness=1.0)
+    #visuals = trimesh.visual.TextureVisuals(uv=uvs, image=rgba_pil, material=material)
+    #mesh.append(trimesh.Trimesh(vertices=verts, faces=triangles, vertex_normals=normals, visual=visuals, validate=True, process=False))
     #print('triangles - ok')
     
-    #ln = []
-    #cr = []
-    #k = 1
-    #kold = 0
-    #c_old = [128,128,128,0]
-    #j=0
-    #while j < gdepth.shape[1]:
-        #i=0
-        #k=j+1
-        #while i < gdepth.shape[0]:
-            #if gdepth[i][j] < 255 and kold != 0 and k != j+1 and (c_old == image[fnum][0][i][j]).all():
-                    #ln.append(trimesh.path.entities.Line([kold, k]))
-                    #cr.append(image[fnum][0][i][j])
-                    #kold=k
-                    #c_old = image[fnum][0][i][j]
-            #k=k+gdepth.shape[1]
-            #i=i+1
-        #j=j+1
+    ln = []
+    cr = []
+    k = 1
+    while k < len(verts):
+        ln.append(k)
+        cr.append(clrs[k])
+        k=k+1
 
     #print(len(ln))
     #print(len(cr))
-    #mesh.append(trimesh.path.path.Path3D(entities=ln, vertices=verts, colors=cr))
+    mesh.append(trimesh.path.path.Path3D(entities=[trimesh.path.entities.Line(ln)], vertices=verts, colors=[[0,0,255,255]]))
       #mesh.append(trimesh.PointCloud(verts, colors=clrs))
       #mesh[len(mesh)-1].merge_vertices()
       #mesh_n.append(str(fnum))