Spaces:
Running
on
Zero
Running
on
Zero
hengli
commited on
Commit
·
f84c11a
1
Parent(s):
d3c7459
update hf link
Browse files- app.py +1 -0
- visual_util.py +2 -0
app.py
CHANGED
|
@@ -125,6 +125,7 @@ def run_model(target_dir, model, anchor_size=100) -> dict:
|
|
| 125 |
] # a
|
| 126 |
del predictions_s
|
| 127 |
|
|
|
|
| 128 |
# Convert tensors to numpy
|
| 129 |
for key in predictions.keys():
|
| 130 |
if isinstance(predictions[key], torch.Tensor):
|
|
|
|
| 125 |
] # a
|
| 126 |
del predictions_s
|
| 127 |
|
| 128 |
+
|
| 129 |
# Convert tensors to numpy
|
| 130 |
for key in predictions.keys():
|
| 131 |
if isinstance(predictions[key], torch.Tensor):
|
visual_util.py
CHANGED
|
@@ -68,6 +68,8 @@ def predictions_to_glb(
|
|
| 68 |
except (ValueError, IndexError):
|
| 69 |
pass
|
| 70 |
|
|
|
|
|
|
|
| 71 |
if "Pointmap" in prediction_mode:
|
| 72 |
print("Using Pointmap Branch")
|
| 73 |
if "world_points" in predictions:
|
|
|
|
| 68 |
except (ValueError, IndexError):
|
| 69 |
pass
|
| 70 |
|
| 71 |
+
for k,v in predictions.items():
|
| 72 |
+
print(f"{k} has {np.isnan(v).sum()} nan values")
|
| 73 |
if "Pointmap" in prediction_mode:
|
| 74 |
print("Using Pointmap Branch")
|
| 75 |
if "world_points" in predictions:
|