Spaces:
Running
on
Zero
Running
on
Zero
fix output of inference
Browse files
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():
|