Axes3D

utils.py

@@ -6,6 +6,7 @@ import numpy as np
 import pandas as pd
 from inference import DepthPredictor
 import matplotlib.pyplot as plt
+from mpl_toolkits.mplot3d import Axes3D
 
 def create_3d_obj(rgb_image, depth_image, depth=10, path='./image.gltf'):
     depth_o3d = o3d.geometry.Image(depth_image)
@@ -180,6 +181,6 @@ def PCL3(image):
     # Step 6: Create a 3D scatter plot using Plotly Express
     #fig = px.scatter_3d(df, x='x', y='y', z='z', color='red', size_max=0.1)
     fig = plt.figure()
-    ax = fig.add_subplot(projection='3d')
+    ax = Axes3D(fig)
     ax.scatter(points[:, 0], points[:, 1], points[:, 2], marker='o', s=0.01, c=colors)
     return fig