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

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
  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())
 
47
 
48
  predicted_depth = model(input_tensor).predicted_depth
49
  depth_map = predicted_depth.squeeze().cpu().numpy()
50
+ num_bins = 1000
51
+ depth_map = np.digitize(depth_map, bins=np.linspace(depth_map.min(), depth_map.max(), num_bins)) - 1
52
  unique_values = np.unique(depth_map)
53
  print(f"Number of unique depth values: {len(unique_values)}")
54
  depth_map = (depth_map - depth_map.min()) / (depth_map.max() - depth_map.min())