init project

modules/pe3r/demo.py

@@ -437,17 +437,17 @@ 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)
-    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)
+    # 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)
+    # 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)
 
     if len(imgs) == 1:
         imgs = [imgs[0], copy.deepcopy(imgs[0])]