update

2026-05-14 17:30:20 +08:00
parent e43171521d
commit 5e6d8046e1
10 changed files with 224 additions and 77 deletions
@@ -6,8 +6,16 @@ import numpy as np
 from PIL import Image


-def depth_to_png16_bytes(depth: np.ndarray) -> bytes:
-    depth = np.asarray(depth, dtype=np.float32)
+def depth_to_png16_bytes(depth: np.ndarray, target_hw: tuple[int, int] | None = None) -> bytes:
+    depth = np.asarray(depth, dtype=np.float32).squeeze()
+    if depth.ndim != 2:
+        raise ValueError(f"depth 应为二维数组，实际 shape={depth.shape}")
+    if target_hw is not None:
+        th, tw = int(target_hw[0]), int(target_hw[1])
+        if th > 0 and tw > 0 and (depth.shape[0] != th or depth.shape[1] != tw):
+            im = Image.fromarray(depth, mode="F")
+            im = im.resize((tw, th), Image.Resampling.BILINEAR)
+            depth = np.asarray(im, dtype=np.float32)
    dmin = float(depth.min())
    dmax = float(depth.max())
    if dmax > dmin: