1

app.py

@@ -82,7 +82,7 @@ def generate_monocular_depth_maps(img_list, depth_prior_name):
     depth_list = []
     focallength_px_list = []
     
-    if depth_prior_name=='depthpro':
+    if depth_prior_name=='Depth Pro':
         model, transform = depth_pro.create_model_and_transforms(device='cuda')
         model.eval()
 
@@ -97,7 +97,7 @@ def generate_monocular_depth_maps(img_list, depth_prior_name):
           depth_list.append(depth)
           focallength_px_list.append(focallength_px)
           #np.savez_compressed(path_depthpro, depth=depth, focallength_px=prediction["focallength_px"].cpu())  
-    elif depth_prior_name=='depthanything':
+    elif depth_prior_name=='Depth Anything V2':
         pipe = pipeline(task="depth-estimation", model="depth-anything/Depth-Anything-V2-Large-hf",device='cuda')
         for image_path in tqdm(img_list):
           #path_depthanything = image_path.replace('.png','_pred_depth_depthanything.npz').replace('.jpg','_pred_depth_depthanything.npz')