init project

app.py

@@ -498,26 +498,8 @@ def get_reconstructed_scene(outdir, filelist, schedule, niter, min_conf_thr,
     outfile = get_3D_model_from_scene(outdir, scene, min_conf_thr, as_pointcloud, mask_sky,
                                       clean_depth, transparent_cams, cam_size)
     torch.cuda.empty_cache()
-    # also return rgb, depth and confidence imgs
-    # depth is normalized with the max value for all images
-    # we apply the jet colormap on the confidence maps
-    # rgbimg = scene.imgs
-    # depths = to_numpy(scene.get_depthmaps())
-    # confs = to_numpy([c for c in scene.im_conf])
-    # # confs = to_numpy([c for c in scene.conf_2])
-    # cmap = pl.get_cmap('jet')
-    # depths_max = max([d.max() for d in depths])
-    # depths = [d / depths_max for d in depths]
-    # confs_max = max([d.max() for d in confs])
-    # confs = [cmap(d / confs_max) for d in confs]
-
-    # imgs = []
-    # for i in range(len(rgbimg)):
-    #     imgs.append(rgbimg[i])
-    #     imgs.append(rgb(depths[i]))
-    #     imgs.append(rgb(confs[i]))
-
-    return scene, outfile #, imgs
+
+    return scene, outfile 
 
 # @spaces.GPU(duration=180)
 # def get_3D_object_from_scene(outdir, text, threshold, scene, min_conf_thr, as_pointcloud, 
@@ -536,27 +518,6 @@ def get_reconstructed_scene(outdir, filelist, schedule, niter, min_conf_thr,
 #     return outfile
 
 
-def set_scenegraph_options(inputfiles, winsize, refid, scenegraph_type):
-    num_files = len(inputfiles) if inputfiles is not None else 1
-    max_winsize = max(1, math.ceil((num_files - 1) / 2))
-    if scenegraph_type == "swin":
-        winsize = gradio.Slider(label="Scene Graph: Window Size", value=max_winsize,
-                                minimum=1, maximum=max_winsize, step=1, visible=True)
-        refid = gradio.Slider(label="Scene Graph: Id", value=0, minimum=0,
-                              maximum=num_files - 1, step=1, visible=False)
-    elif scenegraph_type == "oneref":
-        winsize = gradio.Slider(label="Scene Graph: Window Size", value=max_winsize,
-                                minimum=1, maximum=max_winsize, step=1, visible=False)
-        refid = gradio.Slider(label="Scene Graph: Id", value=0, minimum=0,
-                              maximum=num_files - 1, step=1, visible=True)
-    else:
-        winsize = gradio.Slider(label="Scene Graph: Window Size", value=max_winsize,
-                                minimum=1, maximum=max_winsize, step=1, visible=False)
-        refid = gradio.Slider(label="Scene Graph: Id", value=0, minimum=0,
-                              maximum=num_files - 1, step=1, visible=False)
-    return winsize, refid
-
-
 with tempfile.TemporaryDirectory(suffix='pe3r_gradio_demo') as tmpdirname:
     recon_fun = functools.partial(get_reconstructed_scene, tmpdirname)
     # model_from_scene_fun = functools.partial(get_3D_model_from_scene, tmpdirname)
@@ -607,45 +568,14 @@ with tempfile.TemporaryDirectory(suffix='pe3r_gradio_demo') as tmpdirname:
             # find_btn = gradio.Button("Find")
 
             outmodel = gradio.Model3D()
-            # outgallery = gradio.Gallery(label='rgb,depth,confidence', columns=3, height="100%",
-            #                             visible=False)
-
             # events
-            scenegraph_type.change(set_scenegraph_options,
-                                    inputs=[inputfiles, winsize, refid, scenegraph_type],
-                                    outputs=[winsize, refid])
-            inputfiles.change(set_scenegraph_options,
-                                inputs=[inputfiles, winsize, refid, scenegraph_type],
-                                outputs=[winsize, refid])
+
             run_btn.click(fn=recon_fun,
                             inputs=[inputfiles, schedule, niter, min_conf_thr, as_pointcloud,
                                     mask_sky, clean_depth, transparent_cams, cam_size,
                                     scenegraph_type, winsize, refid],
                             outputs=[scene, outmodel]) # , outgallery
-            # min_conf_thr.release(fn=model_from_scene_fun,
-            #                         inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
-            #                                 clean_depth, transparent_cams, cam_size],
-            #                         outputs=outmodel)
-            # cam_size.change(fn=model_from_scene_fun,
-            #                 inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
-            #                         clean_depth, transparent_cams, cam_size],
-            #                 outputs=outmodel)
-            # as_pointcloud.change(fn=model_from_scene_fun,
-            #                         inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
-            #                                 clean_depth, transparent_cams, cam_size],
-            #                         outputs=outmodel)
-            # mask_sky.change(fn=model_from_scene_fun,
-            #                 inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
-            #                         clean_depth, transparent_cams, cam_size],
-            #                 outputs=outmodel)
-            # clean_depth.change(fn=model_from_scene_fun,
-            #                     inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
-            #                             clean_depth, transparent_cams, cam_size],
-            #                     outputs=outmodel)
-            # transparent_cams.change(model_from_scene_fun,
-            #                         inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
-            #                                 clean_depth, transparent_cams, cam_size],
-            #                         outputs=outmodel)
+            
             # find_btn.click(fn=get_3D_object_from_scene_fun,
             #                     inputs=[text_input, threshold, scene, min_conf_thr, as_pointcloud, mask_sky,
             #                             clean_depth, transparent_cams, cam_size],