huntrezz commited on
Commit
991eddd
·
verified ·
1 Parent(s): 9f5442d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -8
app.py CHANGED
@@ -42,19 +42,14 @@ def plot_depth_map(depth_map, original_image):
42
  ax = fig.add_subplot(111, projection='3d')
43
  x, y = np.meshgrid(range(depth_map.shape[1]), range(depth_map.shape[0]))
44
 
45
- # Flip x and y to mirror the plot
46
- x = np.flip(x, axis=1)
47
- y = np.flip(y, axis=0)
48
-
49
- # Rotate by 90 degrees
50
- x, y = y, x
51
-
52
  # Resize original image to match depth map dimensions
53
  original_image_resized = cv2.resize(original_image, (depth_map.shape[1], depth_map.shape[0]))
54
  colors = original_image_resized.reshape(depth_map.shape[0], depth_map.shape[1], 3) / 255.0
55
 
56
  ax.plot_surface(x, y, depth_map, facecolors=colors, shade=False)
57
- ax.view_init(elev=70, azim=0) # Tilt by 20 degrees (90 - 20 = 70)
 
 
58
  ax.set_zlim(0, 1)
59
  plt.axis('off')
60
  plt.close(fig)
 
42
  ax = fig.add_subplot(111, projection='3d')
43
  x, y = np.meshgrid(range(depth_map.shape[1]), range(depth_map.shape[0]))
44
 
 
 
 
 
 
 
 
45
  # Resize original image to match depth map dimensions
46
  original_image_resized = cv2.resize(original_image, (depth_map.shape[1], depth_map.shape[0]))
47
  colors = original_image_resized.reshape(depth_map.shape[0], depth_map.shape[1], 3) / 255.0
48
 
49
  ax.plot_surface(x, y, depth_map, facecolors=colors, shade=False)
50
+
51
+ # Adjust view to look down at an angle
52
+ ax.view_init(elev=110, azim=180) # Adjusted elevation and azimuth for better view
53
  ax.set_zlim(0, 1)
54
  plt.axis('off')
55
  plt.close(fig)