Remove scene_state from updating

demo/mast3r_demo.py

@@ -110,7 +110,7 @@ def _convert_scene_output_to_glb(outfile, imgs, pts3d, mask, focals, cams2world,
     return outfile
 
 
-def get_3D_model_from_scene(scene_state, min_conf_thr=2, as_pointcloud=False, mask_sky=False,
+def get_3D_model_from_scene(scene, scene_state, min_conf_thr=2, as_pointcloud=False, mask_sky=False,
                             clean_depth=False, transparent_cams=False, cam_size=0.05, TSDF_thresh=0):
     """
     extract 3D_model (glb file) from a reconstructed scene
@@ -121,8 +121,8 @@ def get_3D_model_from_scene(scene_state, min_conf_thr=2, as_pointcloud=False, ma
     if outfile is None:
         return None
 
-    # get optimized values from scene
-    scene = scene_state.sparse_ga
+    # # get optimized values from scene
+    # scene = scenescene_state.sparse_ga
     rgbimg = scene.imgs
     focals = scene.get_focals().cpu()
     cams2world = scene.get_im_poses().cpu()
@@ -232,7 +232,7 @@ def get_reconstructed_scene(image_size, current_scene_state,
         outfile_name = tempfile.mktemp(suffix='_scene.glb', dir=CACHE_PATH)
 
     scene_state = SparseGAState(cache_dir, outfile_name)
-    outfile = get_3D_model_from_scene(scene_state, min_conf_thr, as_pointcloud, mask_sky,
+    outfile = get_3D_model_from_scene(scene, scene_state, min_conf_thr, as_pointcloud, mask_sky,
                                       clean_depth, transparent_cams, cam_size, TSDF_thresh)
     print(f"colmap_data_dir: {colmap_data_dir}")
     print(f"outfile_name: {outfile_name}")
@@ -349,35 +349,35 @@ def mast3r_demo_tab():
                         win_cyclic, refid, TSDF_thresh, shared_intrinsics],
                 outputs=[scene, outmodel]
             )
-            min_conf_thr.release(
-                fn=get_3D_model_from_scene,
-                inputs=[scene, min_conf_thr, as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size, TSDF_thresh],
-                outputs=outmodel
-            )
-            cam_size.change(fn=get_3D_model_from_scene,
-                            inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
-                                    clean_depth, transparent_cams, cam_size, TSDF_thresh],
-                            outputs=outmodel)
-            TSDF_thresh.change(fn=get_3D_model_from_scene,
-                               inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
-                                       clean_depth, transparent_cams, cam_size, TSDF_thresh],
-                               outputs=outmodel)
-            as_pointcloud.change(fn=get_3D_model_from_scene,
-                                 inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
-                                         clean_depth, transparent_cams, cam_size, TSDF_thresh],
-                                 outputs=outmodel)
-            mask_sky.change(fn=get_3D_model_from_scene,
-                            inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
-                                    clean_depth, transparent_cams, cam_size, TSDF_thresh],
-                            outputs=outmodel)
-            clean_depth.change(fn=get_3D_model_from_scene,
-                               inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
-                                       clean_depth, transparent_cams, cam_size, TSDF_thresh],
-                               outputs=outmodel)
-            transparent_cams.change(fn=get_3D_model_from_scene,
-                                    inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
-                                            clean_depth, transparent_cams, cam_size, TSDF_thresh],
-                                    outputs=outmodel)
+            # min_conf_thr.release(
+            #     fn=get_3D_model_from_scene,
+            #     inputs=[scene, min_conf_thr, as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size, TSDF_thresh],
+            #     outputs=outmodel
+            # )
+            # cam_size.change(fn=get_3D_model_from_scene,
+            #                 inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
+            #                         clean_depth, transparent_cams, cam_size, TSDF_thresh],
+            #                 outputs=outmodel)
+            # TSDF_thresh.change(fn=get_3D_model_from_scene,
+            #                    inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
+            #                            clean_depth, transparent_cams, cam_size, TSDF_thresh],
+            #                    outputs=outmodel)
+            # as_pointcloud.change(fn=get_3D_model_from_scene,
+            #                      inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
+            #                              clean_depth, transparent_cams, cam_size, TSDF_thresh],
+            #                      outputs=outmodel)
+            # mask_sky.change(fn=get_3D_model_from_scene,
+            #                 inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
+            #                         clean_depth, transparent_cams, cam_size, TSDF_thresh],
+            #                 outputs=outmodel)
+            # clean_depth.change(fn=get_3D_model_from_scene,
+            #                    inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
+            #                            clean_depth, transparent_cams, cam_size, TSDF_thresh],
+            #                    outputs=outmodel)
+            # transparent_cams.change(fn=get_3D_model_from_scene,
+            #                         inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
+            #                                 clean_depth, transparent_cams, cam_size, TSDF_thresh],
+            #                         outputs=outmodel)
             
     return demo