Add application file

app.py

@@ -31,7 +31,7 @@ def readRAW(path):
 
     return output
 
-def load_bin(file):
+def load_bin(file, threshold=3):
     raw_hist = readRAW(file.name)
 
     multishot = (raw_hist[..., 254] * 1024 + raw_hist[..., 255]).astype(np.float32)
@@ -43,7 +43,7 @@ def load_bin(file):
     img_uint8 = (norm_img * 255).astype(np.uint8)
     img_zoomed = np.repeat(np.repeat(img_uint8, 4, axis=0), 4, axis=1)
 
-    depth_slider_imgs = plot_depth(nor_hist)  # 👈 直接在这里计算
+    depth_slider_imgs = plot_depth(nor_hist,threshold)  # 👈 直接在这里计算
 
     return img_zoomed, raw_hist, nor_hist, depth_slider_imgs
 
@@ -71,7 +71,7 @@ def to_uint8_image(arr):
     norm = (arr) / (np.max(arr) + 1e-8)
     return (norm * 255).astype(np.uint8)
 
-def plot_depth(nor_hist):
+def plot_depth(nor_hist,, threshold):
     kernel = np.ones((3,3))
     output = np.zeros((96, 240, 254))
 
@@ -103,7 +103,7 @@ def plot_depth(nor_hist):
     img_ref = to_uint8_image(ref)
     img_ref_filter = to_uint8_image(ref_filter)
 
-    mask = ref>3
+    mask = ref>threshold
     # 转uint8图像方便展示
     img_tof = to_uint8_image(tof)
     img_filter = to_uint8_image(tof_filter)
@@ -123,7 +123,7 @@ with gr.Blocks() as demo:
     image_display = gr.Image(interactive=True, label="点击像素显示强度曲线")
     histogram = gr.Plot(label="像素强度曲线")
     depth_image_slider = ImageSlider(label="Filter Depth Map with Slider View", elem_id='img-display-output', position=0.5)
-
+    threshold_slider = gr.Slider(1, 30, value=3, step=1, label="Mask 阈值设定 (ref > x)")
 
     raw_hist = gr.State()
     nor_hist = gr.State()
@@ -131,7 +131,7 @@ with gr.Blocks() as demo:
     # 单一入口统一触发
     file_input.change(
         load_bin,
-        inputs=file_input,
+        inputs=[file_input,threshold_slider],
         outputs=[image_display, raw_hist, nor_hist, depth_image_slider]
     )
 
@@ -140,5 +140,10 @@ with gr.Blocks() as demo:
         inputs=[raw_hist, nor_hist],
         outputs=histogram
     )
+    threshold_slider.change(    
+        load_bin,
+        inputs=[file_input,threshold_slider],
+        outputs=[image_display, raw_hist, nor_hist, depth_image_slider])
+
 
 demo.launch()