huntrezz commited on
Commit
ef1c5ec
·
verified ·
1 Parent(s): fd26002

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -47,6 +47,8 @@ def process_frame(image):
47
 
48
  predicted_depth = model(input_tensor).predicted_depth
49
  depth_map = predicted_depth.squeeze().cpu().numpy()
 
 
50
  depth_map = (depth_map - depth_map.min()) / (depth_map.max() - depth_map.min())
51
  depth_map = (depth_map * 255).astype(np.uint8)
52
  depth_map_colored = cv2.applyColorMap(depth_map, cv2.COLORMAP_INFERNO)
 
47
 
48
  predicted_depth = model(input_tensor).predicted_depth
49
  depth_map = predicted_depth.squeeze().cpu().numpy()
50
+ unique_values = np.unique(depth_map)
51
+ print(f"Number of unique depth values: {len(unique_values)}")
52
  depth_map = (depth_map - depth_map.min()) / (depth_map.max() - depth_map.min())
53
  depth_map = (depth_map * 255).astype(np.uint8)
54
  depth_map_colored = cv2.applyColorMap(depth_map, cv2.COLORMAP_INFERNO)