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