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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -47,8 +47,8 @@ def process_frame(image):
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)
 
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)