init project

app.py

@@ -434,7 +434,7 @@ def get_cog_feats(images, pe3r, device):
 
     return cog_seg_maps, rev_cog_seg_maps, multi_view_clip_feats
 
-# @spaces.GPU(duration=180)
+@spaces.GPU(duration=180)
 def get_reconstructed_scene(outdir, pe3r, device, silent, filelist, schedule, niter, min_conf_thr,
                             as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size,
                             scenegraph_type, winsize, refid):
@@ -448,16 +448,16 @@ def get_reconstructed_scene(outdir, pe3r, device, silent, filelist, schedule, ni
     images = Images(filelist=filelist, device=device)
     
     # try:
-    cog_seg_maps, rev_cog_seg_maps, cog_feats = get_cog_feats(images, pe3r, device)
-    imgs = load_images(images, rev_cog_seg_maps, size=512, verbose=not silent)
+    # cog_seg_maps, rev_cog_seg_maps, cog_feats = get_cog_feats(images, pe3r, device)
+    # imgs = load_images(images, rev_cog_seg_maps, size=512, verbose=not silent)
     # except Exception as e:
-    #     rev_cog_seg_maps = []
-    #     for tmp_img in images.np_images:
-    #         rev_seg_map = -np.ones(tmp_img.shape[:2], dtype=np.int64)
-    #         rev_cog_seg_maps.append(rev_seg_map)
-    #     cog_seg_maps = rev_cog_seg_maps
-    #     cog_feats = torch.zeros((1, 1024))
-    #     imgs = load_images(images, rev_cog_seg_maps, size=512, verbose=not silent)
+    rev_cog_seg_maps = []
+    for tmp_img in images.np_images:
+        rev_seg_map = -np.ones(tmp_img.shape[:2], dtype=np.int64)
+        rev_cog_seg_maps.append(rev_seg_map)
+    cog_seg_maps = rev_cog_seg_maps
+    cog_feats = torch.zeros((1, 1024))
+    imgs = load_images(images, rev_cog_seg_maps, size=512, verbose=not silent)
 
     if len(imgs) == 1:
         imgs = [imgs[0], copy.deepcopy(imgs[0])]