Hugging Face's logo

Spaces:
huntrezz
/
RealtimeMonocularDepth
Runtime error

App Files Community
1
huntrezz commited on
Commit
51f6b27
·
verified ·
1 Parent(s): 532c480

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ def process_frame(image):
53
  depth_map = (depth_map * 255).byte()
54
 
55
  # Apply color map
56
- depth_map_colored = color_map[depth_map].squeeze()
57
 
58
  return cv2.cvtColor(depth_map_colored.cpu().numpy(), cv2.COLOR_BGR2RGB)
59
 
 
53
  depth_map = (depth_map * 255).byte()
54
 
55
  # Apply color map
56
+ depth_map_colored = color_map[depth_map.long()]
57
 
58
  return cv2.cvtColor(depth_map_colored.cpu().numpy(), cv2.COLOR_BGR2RGB)
59