global: debug inference

This commit is contained in:
2024-11-01 22:51:16 +00:00
parent 982a3b9b60
commit 287983277a
3 changed files with 10 additions and 2 deletions

View File

@@ -24,8 +24,6 @@ class DataLoadUtil:
for channel in float_channels:
channel_data = exr_file.channel(channel)
img_data.append(np.frombuffer(channel_data, dtype=np.float16).reshape((height, width)))
# 将各通道组合成一个 (height, width, 3) 的 RGB 图像
img = np.stack(img_data, axis=-1)
return img