Update app.py

app.py

@@ -228,39 +228,48 @@ def pano_depth_to_world_points(depth, scale, offset):
 def rgb2gray(rgb):
     return np.dot(rgb[...,:3], [0.333, 0.333, 0.333])
 
-def get_mesh(image, depth, blur_data, scale):
-    fnum = frame_selected
-    offset = locations[fnum]
-    blur_img = blur_image(image[fnum][0], depth[fnum][0], blur_data)
+def get_mesh(image, depth, blur_data, scale, loadall):
+    b = frame_selected
+    d = frame_selected+1
+
+    if loadall == True:
+        b = 0
+        d = len(image)-1
+
+    mesh = []
+    for fnum in range(b, d):
+        offset = locations[fnum]
+        blur_img = blur_image(image[fnum][0], depth[fnum][0], blur_data)
     
-    gdepth = rgb2gray(depth[fnum][0])
-    print('depth to gray - ok')
-    points = pano_depth_to_world_points(gdepth, scale, offset)
-    pts3d = points[0]
-    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
-
-    verts = pts3d.reshape(-1, 3)
-    #triangles = create_triangles(image.shape[0], image.shape[1])
-    #print('triangles - ok')
-    rgba = cv2.cvtColor(blur_img, cv2.COLOR_RGB2RGBA)
-    colors = rgba.reshape(-1, 4)
-    clrs = [[128, 128, 128, 0]]
-
-    for i in range(0,1): #(0,4)
-        clrs = np.concatenate((clrs, colors), axis=0)
-        i = i+1
-
-    #mesh = trimesh.Trimesh(vertices=verts, faces=triangles, vertex_colors=colors)
-    mesh = trimesh.PointCloud(verts, colors=clrs)
-    #material = trimesh.visual.texture.SimpleMaterial(image=image)
-    #texture = trimesh.visual.TextureVisuals(uv=uv, image=image, material=material)
-    #mesh.visual = texture
-    scene = trimesh.Scene([mesh])
+        gdepth = rgb2gray(depth[fnum][0])
+        print('depth to gray - ok')
+        points = pano_depth_to_world_points(gdepth, scale, offset)
+        pts3d = points[0]
+        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
+
+        verts = pts3d.reshape(-1, 3)
+        #triangles = create_triangles(image.shape[0], image.shape[1])
+        #print('triangles - ok')
+        rgba = cv2.cvtColor(blur_img, cv2.COLOR_RGB2RGBA)
+        colors = rgba.reshape(-1, 4)
+        clrs = [[128, 128, 128, 0]]
+
+        for i in range(0,1): #(0,4)
+            clrs = np.concatenate((clrs, colors), axis=0)
+            i = i+1
+
+        #mesh = trimesh.Trimesh(vertices=verts, faces=triangles, vertex_colors=colors)
+        mesh.append(trimesh.PointCloud(verts, colors=clrs))
+        #material = trimesh.visual.texture.SimpleMaterial(image=image)
+        #texture = trimesh.visual.TextureVisuals(uv=uv, image=image, material=material)
+        #mesh.visual = texture
+        
+    scene = trimesh.Scene(mesh)
     print('mesh - ok')
 
     # Save as glb
@@ -532,6 +541,7 @@ with gr.Blocks(css=css) as demo:
             with gr.Accordion(label="Locations", open=False):
                 coords = gr.Textbox(value=example_coords, label="Precise coordinates", show_label=False)
                 scale_in = gr.Slider(value=255, minimum=1, maximum=1023, label="Scale")
+                load_all = gr.Checkbox(label="Load all")
             render = gr.Button("Render")
     
     def on_submit(uploaded_video,model_type,coordinates):
@@ -564,7 +574,7 @@ with gr.Blocks(css=css) as demo:
         return output_video_path
 
     submit.click(on_submit, inputs=[input_video, model_type, coords], outputs=[processed_video, processed_zip, output_frame, output_depth])
-    render.click(partial(get_mesh), inputs=[output_frame, output_depth, blur_in, scale_in], outputs=[result])
+    render.click(partial(get_mesh), inputs=[output_frame, output_depth, blur_in, scale_in, load_all], outputs=[result])
 
     example_files = os.listdir('examples')
     example_files.sort()