import cv2
import numpy as np

# 读取深度图像
depth_image_path = 'aucpuncture2point/configs/using_img/depth.png'
depth_image = cv2.imread(depth_image_path, cv2.IMREAD_UNCHANGED)

if depth_image is None:
    raise ValueError(f"无法读取深度图像文件: {depth_image_path}. 请检查文件路径或格式。")

# 将深度图像归一化到 0-255 范围
depth_normalized = cv2.normalize(depth_image, None, 0, 255, cv2.NORM_MINMAX)

# 确保归一化后的图像是 8 位无符号整数
depth_normalized = np.uint8(depth_normalized)

# 确保 depth_normalized 是单通道图像
if len(depth_normalized.shape) == 2:  # 如果是灰度图像
    # 使用伪彩色映射
    depth_colormap = cv2.applyColorMap(depth_normalized, cv2.COLORMAP_JET)
else:
    raise ValueError("depth_normalized 不是单通道图像，无法应用伪彩色映射。")

# 显示伪彩色深度图像
cv2.imshow('Depth Image (Color Map)', depth_colormap)
cv2.waitKey(0)
cv2.destroyAllWindows()