写一段步态能量图转换为RGB图像的代码

以下是一段 Python 代码，可以将步态能量图转换为 RGB 图像：

import numpy as np
import cv2

def gait_energy_map_to_rgb(gait_energy_map):
    # 将步态能量图归一化到 0-255 的范围
    gait_energy_map = (gait_energy_map - np.min(gait_energy_map)) / (np.max(gait_energy_map) - np.min(gait_energy_map)) * 255
    # 将步态能量图转换为 RGB 图像
    rgb_image = cv2.applyColorMap(gait_energy_map.astype(np.uint8), cv2.COLORMAP_JET)
    return rgb_image

这段代码使用了 OpenCV 库中的 `applyColorMap` 函数，将步态能量图转换为伪彩色图像。具体实现过程是先将步态能量图归一化到 0-255 的范围，然后使用 `applyColorMap` 函数将其转换为 RGB 图像。