Add application file

app.py

@@ -5,6 +5,31 @@ from scipy.ndimage import convolve
 from gradio_imageslider import ImageSlider
 import cv2
 
+def readRAW(path):
+
+    arr = np.fromfile(path, dtype=np.int16).reshape(96,240,256)
+    # 将最后一维重塑为 (-1, 2)，其中 -1 自动计算为 128
+    reshaped = arr.reshape(*arr.shape[:-1], -1, 2)
+    # 交换每一对中的两个元素
+    swapped = reshaped[..., :, ::-1]
+    # 恢复原始形状
+    histogram_data = swapped.reshape(arr.shape)
+    # 定义映射顺序：对每组8行进行调换
+    mapping = [0, 4, 1, 5, 2, 6, 3, 7]
+    # 每组包含的行数
+    group_size = 8
+    num_groups = 12  # 96/8
+
+    # 创建一个用于存储结果的数组（也可以原地修改）
+    output = np.empty_like(histogram_data)
+
+    # 对每个 group 分别进行行重排
+    for g in range(num_groups):
+        start = g * group_size
+        end = start + group_size
+        output[start:end,:,:] = histogram_data[start:end,:,:][mapping,:,:]
+
+    return output
 
 def load_bin(file):
     raw_hist = readRAW(file.name)
@@ -75,39 +100,13 @@ def plot_depth(nor_hist):
     mask = ref>3
     # 转uint8图像方便展示
     img_tof = to_uint8_image(tof)
-    img_filter = to_uint8_image(tof_filter) * mask
+    img_filter = to_uint8_image(tof_filter)
 
     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]
 
-def readRAW(path):
-
-    arr = np.fromfile(path, dtype=np.int16).reshape(96,240,256)
-    # 将最后一维重塑为 (-1, 2)，其中 -1 自动计算为 128
-    reshaped = arr.reshape(*arr.shape[:-1], -1, 2)
-    # 交换每一对中的两个元素
-    swapped = reshaped[..., :, ::-1]
-    # 恢复原始形状
-    histogram_data = swapped.reshape(arr.shape)
-    # 定义映射顺序：对每组8行进行调换
-    mapping = [0, 4, 1, 5, 2, 6, 3, 7]
-    # 每组包含的行数
-    group_size = 8
-    num_groups = 12  # 96/8
-
-    # 创建一个用于存储结果的数组（也可以原地修改）
-    output = np.empty_like(histogram_data)
-
-    # 对每个 group 分别进行行重排
-    for g in range(num_groups):
-        start = g * group_size
-        end = start + group_size
-        output[start:end,:,:] = histogram_data[start:end,:,:][mapping,:,:]
-
-    return output
-
 
 with gr.Blocks() as demo:
     gr.Markdown("## 上传 96×240×256 int16 `.bin/.raw` 文件，点击图像像素查看该像素的 256 帧直方图")