move depth to cpu

2024-10-07 00:46:13 -04:00
parent 2d04f5fcd6
commit d87dcefcf7
1 changed files with 1 additions and 1 deletions
@@ -93,7 +93,7 @@ class DepthPro:
        focal_list = focal_px
        focal_avg = np.mean(focal_px)
        
-        return (depth, focal_list, focal_avg)
+        return (depth.to("cpu"), focal_list, focal_avg)


 class MetricDepthToRelative: