attempt to log frames

app.py

@@ -245,26 +245,28 @@ def _make3d(output_queue: SimpleQueue, images: Image.Image):
         print(f'type(planes)={type(planes)}')
 
         # # get video
-        # chunk_size = 20 if IS_FLEXICUBES else 1
-        # render_size = 384
+        chunk_size = 20 if IS_FLEXICUBES else 1
+        render_size = 384
+
+        print(f'IS_FLEXICUBES={IS_FLEXICUBES}')
         
-        # # frames = []
-        # for i in tqdm(range(0, render_cameras.shape[1], chunk_size)):
-        #     if IS_FLEXICUBES:
-        #         frame = model.forward_geometry(
-        #             planes,
-        #             render_cameras[:, i:i+chunk_size],
-        #             render_size=render_size,
-        #         )['img']
-        #     else:
-        #         frame = model.synthesizer(
-        #             planes,
-        #             cameras=render_cameras[:, i:i+chunk_size],
-        #             render_size=render_size,
-        #         )['images_rgb']
-
-        #     print(f'type(frame)={type(frame)}')
-        #     output_queue.put(("log", "3dvideo", rr.Image(frame)))
+        # frames = []
+        for i in tqdm(range(0, render_cameras.shape[1], chunk_size)):
+            if IS_FLEXICUBES:
+                frame = model.forward_geometry(
+                    planes,
+                    render_cameras[:, i:i+chunk_size],
+                    render_size=render_size,
+                )['img']
+            else:
+                frame = model.synthesizer(
+                    planes,
+                    cameras=render_cameras[:, i:i+chunk_size],
+                    render_size=render_size,
+                )['images_rgb']
+
+            print(f'type(frame)={type(frame)}')
+            output_queue.put(("log", "3dvideo", rr.Image(frame)))
             # frames.append(frame)
 
         # frames = torch.cat(frames, dim=1)
@@ -285,16 +287,8 @@ def _make3d(output_queue: SimpleQueue, images: Image.Image):
         )
         print(f'type(mesh_out)={type(mesh_out)}')
 
-
         vertices, faces, vertex_colors = mesh_out
-        vertices = vertices[:, [1, 2, 0]]
-        print(f'type(vertices)={type(vertices)}')
-        print(f'type(faces)={type(faces)}')
-        print(f'type(vertex_colors)={type(vertex_colors)}')
 
-        save_glb(vertices, faces, vertex_colors, mesh_glb_fpath)
-        save_obj(vertices, faces, vertex_colors, mesh_fpath)
-        
         output_queue.put(
             (
                 "log", 
@@ -307,6 +301,15 @@ def _make3d(output_queue: SimpleQueue, images: Image.Image):
             )
         )
 
+        vertices = vertices[:, [1, 2, 0]]
+
+        print(f'type(vertices)={type(vertices)}')
+        print(f'type(faces)={type(faces)}')
+        print(f'type(vertex_colors)={type(vertex_colors)}')
+
+        save_glb(vertices, faces, vertex_colors, mesh_glb_fpath)
+        save_obj(vertices, faces, vertex_colors, mesh_fpath)
+
         print(f"Mesh saved to {mesh_fpath}")
     
     output_queue.put(("mesh", mesh_out))
@@ -379,7 +382,7 @@ If you have any questions, feel free to open a discussion or contact us at <b>bl
 with gr.Blocks() as demo:
     gr.Markdown(_HEADER_)
     with gr.Row(variant="panel"):
-        with gr.Column():
+        with gr.Column(scale=1):
             with gr.Row():
                 input_image = gr.Image(
                     label="Input Image",
@@ -419,7 +422,7 @@ with gr.Blocks() as demo:
                     examples_per_page=16
                 )
 
-        with gr.Column():
+        with gr.Column(scale=4):
             
             viewer = Rerun(streaming=True, height=800)
 
@@ -467,7 +470,7 @@ with gr.Blocks() as demo:
     #     fn=generate_mvs,
     #     inputs=[processed_image, sample_steps, sample_seed],
     #     outputs=[mv_images, mv_show_images]
-        
+
     # ).success(
     #     fn=make3d,
     #     inputs=[mv_images],