matplotlib

utils.py

@@ -177,10 +177,12 @@ def PCL3(image):
     sizes = np.zeros(colors.shape)
     sizes[:] = 0.01
     colors = [tuple(c) for c in colors]
-    #ax = Axes3D(fig)
-    data = {'x': points[:, 0], 'y': points[:, 1], 'z': points[:, 2], 'sizes': sizes[:, 0]}
-    df = pd.DataFrame(data)
+    fig = plt.figure()
+    ax = fig.add_subplot(111, projection='3d')
+    ax.scatter(points[:, 0], points[:, 1], points[:, 2], c=colors, s=0.01)
+    #data = {'x': points[:, 0], 'y': points[:, 1], 'z': points[:, 2], 'sizes': sizes[:, 0]}
+    #df = pd.DataFrame(data)
     # Step 6: Create a 3D scatter plot using Plotly Express
-    fig = px.scatter_3d(df, x='x', y='y', z='z', color=colors, size="sizes")
+    #fig = px.scatter_3d(df, x='x', y='y', z='z', color=colors, size="sizes")
     
     return fig