init project

app.py

@@ -501,23 +501,23 @@ def get_reconstructed_scene(outdir, filelist, schedule, niter, min_conf_thr,
     # 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
+    # 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
 
 @spaces.GPU(duration=180)
 def get_3D_object_from_scene(outdir, text, threshold, scene, min_conf_thr, as_pointcloud, 
@@ -607,8 +607,8 @@ 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)
+            # outgallery = gradio.Gallery(label='rgb,depth,confidence', columns=3, height="100%",
+            #                             visible=False)
 
             # events
             scenegraph_type.change(set_scenegraph_options,
@@ -621,7 +621,7 @@ with tempfile.TemporaryDirectory(suffix='pe3r_gradio_demo') as tmpdirname:
                             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])
+                            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],