scale depth by 65535/maxdepth

app.py

@@ -175,7 +175,7 @@ with gr.Blocks(css=css) as demo:
     with gr.Row():
         depth_image_slider = ImageSlider(label="Depth Map with Slider View", elem_id='img-display-output', position=0.5)
         
-    raw_file = gr.File(label="16-Bit Raw Depth, Multiplier:256")
+    raw_file = gr.File(label="16-Bit Raw Depth, Multiplier:65535/80")
     submit = gr.Button("Submit")
 
     def on_submit(image, mode, patch_number, resolution_h, resolution_w, patch_split_number_h, patch_split_number_w, color_map):
@@ -202,7 +202,8 @@ with gr.Blocks(css=css) as demo:
         pf_prediction_colored = colorize_depth(pf_prediction, cmap=color_map)
         coarse_predictioncolored = colorize_depth(coarse_prediction, cmap=color_map)
         
-        raw_depth = Image.fromarray((pf_prediction*256).astype('uint16'))
+        max_depth = 80
+        raw_depth = Image.fromarray((pf_prediction*(65535 / max_depth)).astype('uint16'))
         tmp = tempfile.NamedTemporaryFile(suffix='.png', delete=False)
         raw_depth.save(tmp.name)