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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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="/
|
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",
|