hengli commited on
Commit
f84c11a
·
1 Parent(s): d3c7459

update hf link

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. 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: