init project

app.py

@@ -498,26 +498,26 @@ 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)
 
-    # # 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
+    # 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
 
 @spaces.GPU(duration=180)
 def get_3D_object_from_scene(outdir, text, threshold, scene, min_conf_thr, as_pointcloud,