init project

app.py

@@ -256,7 +256,7 @@ def slerp_multiple(vectors, t_values):
     
     return interpolated_vector
 
-@torch.no_grad
+# @torch.no_grad
 def get_mask_from_img_sam1(sam1_image, yolov8_image, original_size, input_size, transform):
 
     device = 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -326,7 +326,7 @@ def get_mask_from_img_sam1(sam1_image, yolov8_image, original_size, input_size,
 
     return ret_mask
 
-@torch.no_grad
+# @torch.no_grad
 def get_cog_feats(images):
 
     device = 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -540,7 +540,7 @@ 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)
     
-    # scene.to('cpu')
+    scene.to('cpu')
     torch.cuda.empty_cache()
 
     return scene, outfile