Add application file

app.py

@@ -99,6 +99,9 @@ def plot_depth(nor_hist):
     ref = np.max(arr, axis=2)
     ref_filter = np.max(arr_expandfilter, axis=2)
 
+    img_ref = to_uint8_image(ref)
+    img_ref_filter = to_uint8_image(ref_filter)
+
     mask = ref>3
     # 转uint8图像方便展示
     img_tof = to_uint8_image(tof)
@@ -107,7 +110,8 @@ def plot_depth(nor_hist):
     colored_tof = cv2.applyColorMap(img_tof, cv2.COLORMAP_VIRIDIS)[:, :, ::-1]
     colored_tof_filter = cv2.applyColorMap(img_filter, cv2.COLORMAP_VIRIDIS)[:, :, ::-1]
 
-    return [colored_tof, colored_tof_filter]
+    # return [colored_tof, colored_tof_filter]
+    return [img_ref, img_ref_filter]
 
 
 with gr.Blocks() as demo: