如何使用OpenCV为归一化浮点numpy数组应用颜色映射？

原代码 #

for i in range(arr.shape[0]):
    for j in range(arr.shape[1]):
        arr[i,j] = arr[i,j] * 255
        arr[i,j] = int(round(arr[i,j]))
        
arr.astype(np.uint8)
colorMapImage = cv2.cvtColor(arr,cv2.COLOR_GRAY2BGR)
colorMapImage = cv2.applyColorMap(colorMapImage, cv2.COLORMAP_VIRIDIS)

报错信息 #

[ERROR] [1719390004.370940]: Error processing image: OpenCV(4.2.0) /home/myPC/opencv_build/opencv/modules/imgproc/src/colormap.cpp:714: error: (-5:Bad argument) cv::ColorMap only supports source images of type CV_8UC1 or CV_8UC3 in function 'operator()'

补充说明 #

• 转换前数组已归一化至0-1区间
• 数组形状为(300, 300)
• 应用颜色映射前colorMapImage[0,0]结果为[5. 5. 5.]
• 执行arr.dtype()时报错：'numpy.dtype[float32]' object is not callable
• arr和colorMapImage均为numpy.ndarray类型

错误原因 #

1. 数组类型未生效：arr.astype(np.uint8)返回新的uint8数组，但未赋值回变量，原arr仍为float32类型；嵌套循环中把元素转int后，赋值回float数组会自动转为浮点型（如5变5.0），数组类型不变。
2. 多余通道转换：cv2.cvtColor(arr, cv2.COLOR_GRAY2BGR)将单通道浮点数组转为3通道浮点数组，而cv2.applyColorMap仅支持8位单通道/3通道图像，浮点类型不兼容。
3. 属性调用错误：arr.dtype是数组属性，不是方法，不能加括号调用，正确写法是print(arr.dtype)。

修正代码 #

import numpy as np
import cv2

# 假设arr是已归一化到0-1的(300,300) float数组
# arr = ... 

# 一步完成缩放、取整并转换为uint8类型
arr_8bit = (arr * 255).round().astype(np.uint8)

# 直接对单通道8位灰度图应用颜色映射
colorMapImage = cv2.applyColorMap(arr_8bit, cv2.COLORMAP_VIRIDIS)

代码说明 #

• 用NumPy矢量运算替代嵌套循环，效率更高，避免循环中的类型问题。
• (arr * 255).round()将0-1区间值缩放至0-255并取整，astype(np.uint8)转换为8位无符号整数，必须赋值给新变量（或覆盖原变量），因为astype不修改原数组。
• cv2.applyColorMap直接支持单通道8位灰度图，无需先转BGR，简化流程同时满足类型要求。

内容的提问来源于stack exchange，提问作者mericgeren