Spaces:

NN-BRD
/

hackathon_depth_segment

Runtime error

jens commited on Aug 10, 2023

Commit

7bcdb21

1 Parent(s): 6bea180

fix

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -146,7 +146,7 @@ def generate_PCL(image):
     depth_result = depth_predictor.predict(image)
     image = np.array(image)
     pcd = array_PCL(image, depth_result)
-    fig = px.scatter_3d(x=pcd[:, 0], y=pcd[:, 1], z=pcd[:, 2], size_max=0.1)
     return fig

     depth_result = depth_predictor.predict(image)
     image = np.array(image)
     pcd = array_PCL(image, depth_result)
+    fig = px.scatter_3d(x=pcd[:, 0], y=pcd[:, 1], z=pcd[:, 2], size_max=0.01)
     return fig