init project

app.py

@@ -86,7 +86,7 @@ def _convert_scene_output_to_glb(outdir, imgs, pts3d, mask, focals, cams2world,
     scene.export(file_obj=outfile)
     return outfile
 
-# @spaces.GPU(duration=180)
+@spaces.GPU(duration=180)
 def get_3D_model_from_scene(outdir, scene, min_conf_thr=3, as_pointcloud=False, mask_sky=False,
                             clean_depth=False, transparent_cams=False, cam_size=0.05):
     """
@@ -467,11 +467,15 @@ def get_reconstructed_scene(outdir, filelist, schedule, niter, min_conf_thr,
     elif scenegraph_type == "oneref":
         scenegraph_type = scenegraph_type + "-" + str(refid)
 
+    print('0')
     pairs = make_pairs(imgs, scene_graph=scenegraph_type, prefilter=None, symmetrize=True)
+    print('a')
     output = inference(pairs, pe3r.mast3r, device, batch_size=1, verbose=not silent)
+    print('b')
     mode = GlobalAlignerMode.PointCloudOptimizer if len(imgs) > 2 else GlobalAlignerMode.PairViewer
     scene_1 = global_aligner(output, cog_seg_maps, rev_cog_seg_maps, cog_feats, device=device, mode=mode, verbose=not silent)
     lr = 0.01
+    print('c')
     # if mode == GlobalAlignerMode.PointCloudOptimizer:
     loss = scene_1.compute_global_alignment(tune_flg=True, init='mst', niter=niter, schedule=schedule, lr=lr)
 

modules/dust3r/cloud_opt/base_opt.py

@@ -442,7 +442,7 @@ class BasePCOptimizer (nn.Module):
         return refresh_confidence_map.view(H, W)
 
 
-    @torch.cuda.amp.autocast(enabled=False)
+    # @torch.cuda.amp.autocast(enabled=False)
     def compute_global_alignment(self, tune_flg=False, init=None, niter_PnP=10, **kw):
         
         if tune_flg: