midas remote

xiaobb001 · May 14, 2024 · 27a4b28 · 27a4b28
1 parent fccd58c
commit 27a4b28
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/scene/__init__.py b/scene/__init__.py
@@ -81,8 +81,9 @@ def __init__(self, args: ModelParams, gaussians: GaussianModel, load_iteration=N
 
         self.cameras_extent = scene_info.nerf_normalization["radius"]
 
-        midas = torch.hub.load(
-            "/home/cwb/.cache/torch/hub/intel-isl_MiDaS_master", "DPT_Hybrid", source="local")
+        # midas = torch.hub.load(
+        #     "/home/cwb/.cache/torch/hub/intel-isl_MiDaS_master", "DPT_Hybrid", source="local")
+        midas = torch.hub.load("intel-isl/MiDaS", "DPT_Hybrid")
         midas.cuda()
         midas.eval()
         for param in midas.parameters():