写一段图像伪彩色增强的python代码

以下是一个简单的图像伪彩色增强的Python代码示例：

import cv2
import numpy as np

# 加载原始灰度图像
img_gray = cv2.imread("input_gray.jpg", cv2.IMREAD_GRAYSCALE)

# 颜色映射函数
def colormap_fn(val):
    if val < 60:
        return (0, 0, 0)  # 黑色
    elif val < 120:
        return (255, 0, 0)  # 蓝色
    elif val < 180:
        return (0, 255, 0)  # 绿色
    elif val < 240:
        return (0, 0, 255)  # 红色
    else:
        return (255, 255, 255)  # 白色

# 伪彩色增强
img_color = np.zeros((img_gray.shape[0], img_gray.shape[1], 3), dtype=np.uint8)
for i in range(img_gray.shape[0]):
    for j in range(img_gray.shape[1]):
        img_color[i, j] = colormap_fn(img_gray[i, j])

# 保存伪彩色图像
cv2.imwrite("output_color.jpg", img_color)

以上代码中，我们首先加载了一个原始的灰度图像。然后，我们定义了一个颜色映射函数，该函数将根据输入像素的灰度值返回相应的伪彩色值。接下来，我们对每个像素应用了颜色映射函数，生成了一个伪彩色图像。最后，我们将伪彩色图像保存为输出文件。