lch01 commited on
Commit
2762245
·
1 Parent(s): 2e95ac7

fix output of inference

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -122,6 +122,8 @@ def run_model(target_dir, model) -> dict:
122
  extrinsic, intrinsic = pose_encoding_to_extri_intri(predictions["pose_enc"].unsqueeze(0) if predictions["pose_enc"].ndim == 2 else predictions["pose_enc"], images.shape[-2:])
123
  predictions["extrinsic"] = extrinsic
124
  predictions["intrinsic"] = intrinsic
 
 
125
 
126
  # Convert tensors to numpy
127
  for key in predictions.keys():
 
122
  extrinsic, intrinsic = pose_encoding_to_extri_intri(predictions["pose_enc"].unsqueeze(0) if predictions["pose_enc"].ndim == 2 else predictions["pose_enc"], images.shape[-2:])
123
  predictions["extrinsic"] = extrinsic
124
  predictions["intrinsic"] = intrinsic
125
+ print("Extrinsic shape:", predictions["extrinsic"].shape)
126
+ print("Intrinsic shape:", predictions["intrinsic"].shape)
127
 
128
  # Convert tensors to numpy
129
  for key in predictions.keys():