Spaces:
Running
on
Zero
Running
on
Zero
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")
|