app.py
CHANGED
|
@@ -20,6 +20,16 @@ from PIL import Image
|
|
| 20 |
import os.path as path
|
| 21 |
import sys
|
| 22 |
import tempfile
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
from dust3r.inference import inference
|
| 24 |
from dust3r.model import AsymmetricCroCo3DStereo
|
| 25 |
from dust3r.image_pairs import make_pairs
|
|
|
|
| 20 |
import os.path as path
|
| 21 |
import sys
|
| 22 |
import tempfile
|
| 23 |
+
|
| 24 |
+
# Step 1: Navigate to the desired directory
|
| 25 |
+
os.chdir('croco/models/curope/')
|
| 26 |
+
|
| 27 |
+
# Step 2: Execute the setup command
|
| 28 |
+
os.system('python setup.py build_ext --inplace')
|
| 29 |
+
|
| 30 |
+
# Step 3: Return to the original directory
|
| 31 |
+
os.chdir('../../../')
|
| 32 |
+
|
| 33 |
from dust3r.inference import inference
|
| 34 |
from dust3r.model import AsymmetricCroCo3DStereo
|
| 35 |
from dust3r.image_pairs import make_pairs
|
croco/models/__pycache__/pos_embed.cpython-311.pyc
CHANGED
|
Binary files a/croco/models/__pycache__/pos_embed.cpython-311.pyc and b/croco/models/__pycache__/pos_embed.cpython-311.pyc differ
|
|
|
dust3r/utils/__pycache__/image_pose.cpython-311.pyc
CHANGED
|
Binary files a/dust3r/utils/__pycache__/image_pose.cpython-311.pyc and b/dust3r/utils/__pycache__/image_pose.cpython-311.pyc differ
|
|
|
dust3r/utils/image_pose.py
CHANGED
|
@@ -303,7 +303,8 @@ def load_images(folder_or_list, depth_list, focallength_px_list, size, square_ok
|
|
| 303 |
|
| 304 |
if dynamic_mask_root is not None:
|
| 305 |
dynamic_mask_path = os.path.join(dynamic_mask_root, os.path.basename(path))
|
| 306 |
-
|
|
|
|
| 307 |
# dynamic_mask_path = full_path.replace('final', 'dynamic_label_perfect').replace('clean', 'dynamic_label_perfect').replace('MPI-Sintel-training_images','MPI-Sintel-depth-training')
|
| 308 |
#print(dynamic_mask_path)
|
| 309 |
if os.path.exists(dynamic_mask_path):
|
|
|
|
| 303 |
|
| 304 |
if dynamic_mask_root is not None:
|
| 305 |
dynamic_mask_path = os.path.join(dynamic_mask_root, os.path.basename(path))
|
| 306 |
+
else: # Sintel dataset handling
|
| 307 |
+
dynamic_mask_path = ''
|
| 308 |
# dynamic_mask_path = full_path.replace('final', 'dynamic_label_perfect').replace('clean', 'dynamic_label_perfect').replace('MPI-Sintel-training_images','MPI-Sintel-depth-training')
|
| 309 |
#print(dynamic_mask_path)
|
| 310 |
if os.path.exists(dynamic_mask_path):
|