Spaces:
Running
Running
fix double numpy array
Browse files
app.py
CHANGED
|
@@ -45,7 +45,7 @@ def predict(frame):
|
|
| 45 |
image = image.to(device)
|
| 46 |
prediction = model.infer(image)
|
| 47 |
depth = prediction["depth"].squeeze().detach().cpu().numpy()
|
| 48 |
-
return depth
|
| 49 |
|
| 50 |
|
| 51 |
@rr.thread_local_stream("rerun_example_ml_depth_pro")
|
|
|
|
| 45 |
image = image.to(device)
|
| 46 |
prediction = model.infer(image)
|
| 47 |
depth = prediction["depth"].squeeze().detach().cpu().numpy()
|
| 48 |
+
return depth, prediction["focallength_px"].item()
|
| 49 |
|
| 50 |
|
| 51 |
@rr.thread_local_stream("rerun_example_ml_depth_pro")
|