update mast3r module

app.py

@@ -59,9 +59,9 @@ def local_get_reconstructed_scene(filelist, min_conf_thr, matching_conf_thr,
                                   as_pointcloud, cam_size,
                                   shared_intrinsics, **kw):
     lr1 = 0.07
-    niter1 = 500
-    lr2 = 0.014
-    niter2 = 200
+    niter1 = 300
+    lr2 = 0.01
+    niter2 = 300
     optim_level = 'refine'
     mask_sky, clean_depth, transparent_cams = False, True, False
     if len(filelist) < 5:
@@ -74,7 +74,7 @@ def local_get_reconstructed_scene(filelist, min_conf_thr, matching_conf_thr,
         winsize = min(5, max_winsize)
     refid = 0
     win_cyclic = False
-    scene_state, outfile = get_reconstructed_scene(tmpdirname, gradio_delete_cache, model, device, silent, image_size, None,
+    scene_state, outfile = get_reconstructed_scene(tmpdirname, gradio_delete_cache, model, None, device, silent, image_size, None,
                                                    filelist, optim_level, lr1, niter1, lr2, niter2, min_conf_thr, matching_conf_thr,
                                                    as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size, scenegraph_type, winsize,
                                                    win_cyclic, refid, TSDF_thresh=0, shared_intrinsics=shared_intrinsics, **kw)
@@ -87,11 +87,13 @@ def local_get_reconstructed_scene(filelist, min_conf_thr, matching_conf_thr,
 def run_example(snapshot, matching_conf_thr, min_conf_thr, cam_size, as_pointcloud, shared_intrinsics, filelist, **kw):
     return local_get_reconstructed_scene(filelist, min_conf_thr, matching_conf_thr, as_pointcloud, cam_size, shared_intrinsics, **kw)
 
+
 css = """.gradio-container {margin: 0 !important; min-width: 100%};"""
 title = "MASt3R Demo"
 with gradio.Blocks(css=css, title=title, delete_cache=(gradio_delete_cache, gradio_delete_cache)) as demo:
     filestate = gradio.State(None)
-    gradio.HTML('<h2 style="text-align: center;">3D Reconstruction with MASt3R</h2>')
+    gradio.HTML(
+        '<h2 style="text-align: center;">3D Reconstruction with MASt3R</h2>')
     gradio.HTML('<p>Upload one or multiple images (wait for them to be fully uploaded before hitting the run button). '
                 'We tested with up to 18 images before running into the allocation timeout - set at 3 minutes but your mileage may vary. '
                 'At the very bottom of this page, you will find an example. If you click on it, it will pull the 3D reconstruction from 7 images of the small Naver Labs Europe tower from cache. '
@@ -102,13 +104,15 @@ with gradio.Blocks(css=css, title=title, delete_cache=(gradio_delete_cache, grad
         inputfiles = gradio.File(file_count="multiple")
         snapshot = gradio.Image(None, visible=False)
         with gradio.Row():
-            matching_conf_thr = gradio.Slider(label="Matching Confidence Thr", value=2.,
+            matching_conf_thr = gradio.Slider(label="Matching Confidence Thr", value=0.,
                                               minimum=0., maximum=30., step=0.1,
                                               info="Before Fallback to Regr3D!")
             # adjust the confidence threshold
-            min_conf_thr = gradio.Slider(label="min_conf_thr", value=1.5, minimum=0.0, maximum=10, step=0.1)
+            min_conf_thr = gradio.Slider(
+                label="min_conf_thr", value=1.5, minimum=0.0, maximum=10, step=0.1)
             # adjust the camera size in the output pointcloud
-            cam_size = gradio.Slider(label="cam_size", value=0.2, minimum=0.001, maximum=1.0, step=0.001)
+            cam_size = gradio.Slider(
+                label="cam_size", value=0.2, minimum=0.001, maximum=1.0, step=0.001)
         with gradio.Row():
             as_pointcloud = gradio.Checkbox(value=True, label="As pointcloud")
             shared_intrinsics = gradio.Checkbox(value=False, label="Shared intrinsics",
@@ -119,7 +123,8 @@ with gradio.Blocks(css=css, title=title, delete_cache=(gradio_delete_cache, grad
         examples = gradio.Examples(
             examples=[
                 [
-                    os.path.join(HERE_PATH, 'mast3r/assets/NLE_tower/FF5599FD-768B-431A-AB83-BDA5FB44CB9D-83120-000041DADDE35483.jpg'),
+                    os.path.join(
+                        HERE_PATH, 'mast3r/assets/NLE_tower/FF5599FD-768B-431A-AB83-BDA5FB44CB9D-83120-000041DADDE35483.jpg'),
                     0.0, 1.5, 0.2, True, False,
                      [os.path.join(HERE_PATH, 'mast3r/assets/NLE_tower/01D90321-69C8-439F-B0B0-E87E7634741C-83120-000041DAE419D7AE.jpg'),
                       os.path.join(
@@ -135,7 +140,8 @@ with gradio.Blocks(css=css, title=title, delete_cache=(gradio_delete_cache, grad
                       os.path.join(HERE_PATH, 'mast3r/assets/NLE_tower/FF5599FD-768B-431A-AB83-BDA5FB44CB9D-83120-000041DADDE35483.jpg')]
                 ]
             ],
-            inputs=[snapshot, matching_conf_thr, min_conf_thr, cam_size, as_pointcloud, shared_intrinsics, inputfiles],
+            inputs=[snapshot, matching_conf_thr, min_conf_thr,
+                    cam_size, as_pointcloud, shared_intrinsics, inputfiles],
             outputs=[filestate, outmodel],
             fn=run_example,
             cache_examples="lazy",

mast3r

@@ -1 +1 @@
-Subproject commit b587bb36c11351e9303517d3940a0cd3d750cae6
+Subproject commit fb8dfe783043764f6ad78dd021bfcd88bc1d73a2