init project

app.py

@@ -448,7 +448,7 @@ def get_reconstructed_scene(outdir, device, silent, filelist, schedule, niter, m
     images = Images(filelist=filelist, device=device)
     
     # try:
-    cog_seg_maps, rev_cog_seg_maps, cog_feats = get_cog_feats(images, device)
+    # cog_seg_maps, rev_cog_seg_maps, cog_feats = get_cog_feats(images, device)
     # imgs = load_images(images, rev_cog_seg_maps, size=512, verbose=not silent)
     # except Exception as e:
         # rev_cog_seg_maps = []
@@ -459,14 +459,14 @@ def get_reconstructed_scene(outdir, device, silent, filelist, schedule, niter, m
         # 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])]
-        imgs[1]['idx'] = 1
+    # if len(imgs) == 1:
+    #     imgs = [imgs[0], copy.deepcopy(imgs[0])]
+    #     imgs[1]['idx'] = 1
 
-    if scenegraph_type == "swin":
-        scenegraph_type = scenegraph_type + "-" + str(winsize)
-    elif scenegraph_type == "oneref":
-        scenegraph_type = scenegraph_type + "-" + str(refid)
+    # if scenegraph_type == "swin":
+    #     scenegraph_type = scenegraph_type + "-" + str(winsize)
+    # elif scenegraph_type == "oneref":
+    #     scenegraph_type = scenegraph_type + "-" + str(refid)
 
     # pairs = make_pairs(imgs, scene_graph=scenegraph_type, prefilter=None, symmetrize=True)
     # output = inference(pairs, pe3r.mast3r, device, batch_size=1, verbose=not silent)