Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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())
|