cyun9286 commited on
Commit
20addcb
1 Parent(s): d8fa968
app.py CHANGED
@@ -92,7 +92,16 @@ def generate_monocular_depth_maps(img_list, depth_prior_name):
92
  focallength_px_list = []
93
 
94
  if depth_prior_name=='Depth Pro':
95
- model, transform = depth_pro.create_model_and_transforms(device='cuda')
 
 
 
 
 
 
 
 
 
96
  model.eval()
97
 
98
  for image_path in tqdm(img_list):
 
92
  focallength_px_list = []
93
 
94
  if depth_prior_name=='Depth Pro':
95
+
96
+ DEFAULT_MONODEPTH_CONFIG_DICT = DepthProConfig(
97
+ patch_encoder_preset="dinov2l16_384",
98
+ image_encoder_preset="dinov2l16_384",
99
+ checkpoint_uri=f'{HERE_PATH}/third_party/ml-depth-pro/checkpoints/depth_pro.pt',
100
+ decoder_features=256,
101
+ use_fov_head=True,
102
+ fov_encoder_preset='dinov2l16_384',
103
+ )
104
+ model, transform = depth_pro.create_model_and_transforms(config=DEFAULT_MONODEPTH_CONFIG_DICT, device='cuda')
105
  model.eval()
106
 
107
  for image_path in tqdm(img_list):
dust3r/cloud_opt_flow/__pycache__/base_opt.cpython-311.pyc CHANGED
Binary files a/dust3r/cloud_opt_flow/__pycache__/base_opt.cpython-311.pyc and b/dust3r/cloud_opt_flow/__pycache__/base_opt.cpython-311.pyc differ
 
dust3r/cloud_opt_flow/base_opt.py CHANGED
@@ -28,7 +28,7 @@ from PIL import Image
28
  import os.path as path
29
  HERE_PATH = path.normpath(path.dirname(__file__))
30
  print('**',HERE_PATH)
31
-
32
  def c2w_to_tumpose(c2w):
33
  """
34
  Convert a camera-to-world matrix to a tuple of translation and rotation
 
28
  import os.path as path
29
  HERE_PATH = path.normpath(path.dirname(__file__))
30
  print('**',HERE_PATH)
31
+ #** /home/user/app/dust3r/cloud_opt_flow
32
  def c2w_to_tumpose(c2w):
33
  """
34
  Convert a camera-to-world matrix to a tuple of translation and rotation
third_party/ml-depth-pro/src/depth_pro/depth_pro.py CHANGED
@@ -40,7 +40,7 @@ class DepthProConfig:
40
  DEFAULT_MONODEPTH_CONFIG_DICT = DepthProConfig(
41
  patch_encoder_preset="dinov2l16_384",
42
  image_encoder_preset="dinov2l16_384",
43
- checkpoint_uri="/home/lipeng/ljh_code/Video_Depth_CVPR2025-main/ml-depth-pro/checkpoints/depth_pro.pt",
44
  decoder_features=256,
45
  use_fov_head=True,
46
  fov_encoder_preset="dinov2l16_384",
 
40
  DEFAULT_MONODEPTH_CONFIG_DICT = DepthProConfig(
41
  patch_encoder_preset="dinov2l16_384",
42
  image_encoder_preset="dinov2l16_384",
43
+ checkpoint_uri="./third_party/ml-depth-pro/checkpoints/depth_pro.pt",
44
  decoder_features=256,
45
  use_fov_head=True,
46
  fov_encoder_preset="dinov2l16_384",