freealise commited on
Commit
aef558c
·
verified ·
1 Parent(s): 0585f2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -82,7 +82,7 @@ with gr.Blocks(css=css) as demo:
82
 
83
  depth = (depth - depth.min()) / (depth.max() - depth.min()) * 255.0
84
  depth = depth.cpu().numpy().astype(np.uint8)
85
- colored_depth = cv2.applyColorMap(depth, cv2.COLORMAP_INFERNO)[:, :, ::-1]
86
 
87
  return [(original_image, colored_depth), tmp.name]
88
 
 
82
 
83
  depth = (depth - depth.min()) / (depth.max() - depth.min()) * 255.0
84
  depth = depth.cpu().numpy().astype(np.uint8)
85
+ colored_depth = cv2.applyColorMap(depth, cv2.COLORMAP_BONE)[:, :, ::-1]
86
 
87
  return [(original_image, colored_depth), tmp.name]
88