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
@@ -247,10 +247,10 @@ def pano_depth_to_world_points(depth):
             z = radius * np.sin(d_lon) * np.sin(d_lat)
 
             pts = np.stack([x, y, z], axis=1)
-            uvs = np.stack([lon, lat], axis=1)
+            #uvs = np.stack([lon, lat], axis=1)
             
             pts3d = np.concatenate((pts3d, pts), axis=0)
-            uv = np.concatenate((uv, uvs), axis=0)
+            #uv = np.concatenate((uv, uvs), axis=0)
             #print(f'i: {i}, j: {j}')
             j = j+1
         i = i+1
@@ -280,13 +280,13 @@ def get_mesh(image, depth, blur_data, loadall):
     print('depth to gray - ok')
     points = pano_depth_to_world_points(gdepth)
     pts3d = points[0]
-    uv = points[1]
+    #uv = points[1]
     print('radius from depth - ok')
 
     # Create a trimesh mesh from the points
     # Each pixel is connected to its 4 neighbors
     # colors are the RGB values of the image
-    uvs = uv.reshape(-1, 2)
+    #uvs = uv.reshape(-1, 2)
     verts = pts3d.reshape(-1, 3)
     rgba = cv2.cvtColor(blur_img, cv2.COLOR_RGB2RGBA)
     colors = rgba.reshape(-1, 4)
@@ -296,15 +296,15 @@ def get_mesh(image, depth, blur_data, loadall):
         clrs = np.concatenate((clrs, colors), axis=0)
         i = i+1
 
-    triangles = create_triangles(rgba.shape[0], rgba.shape[1])
-    mesh.append(trimesh.Trimesh(vertices=verts, faces=triangles))
-    rgba_pil =  Image.fromarray(rgba.astype(np.uint8))
-    material = trimesh.visual.texture.SimpleMaterial(image=rgba_pil)
-    visuals = trimesh.visual.TextureVisuals(uv=uvs, image=rgba_pil, material=material)
-    mesh[len(mesh)-1].visual = visuals
-    print('triangles - ok')
+    #triangles = create_triangles(rgba.shape[0], rgba.shape[1])
+    #mesh.append(trimesh.Trimesh(vertices=verts, faces=triangles))
+    #rgba_pil =  Image.fromarray(rgba.astype(np.uint8))
+    #material = trimesh.visual.texture.SimpleMaterial(image=rgba_pil)
+    #visuals = trimesh.visual.TextureVisuals(uv=uvs, image=rgba_pil, material=material)
+    #mesh[len(mesh)-1].visual = visuals
+    #print('triangles - ok')
     
-    #mesh.append(trimesh.PointCloud(verts, colors=clrs))
+    mesh.append(trimesh.PointCloud(verts, colors=clrs))
     mesh_n.append(str(fnum))
     mesh[len(mesh)-1].merge_vertices()