cv2.applyColorMap（）使用中出现的错误

cv::ColorMap only supports source images of type CV_8UC1 or CV_8UC3 in function cv2.applyColorMap（）使用中出现的错误

最近使用cv2.cv2.applyColorMap（）将一个2维ndarray转换成伪彩色图片，发现经常爆出“cv::ColorMap only supports source images of type CV_8UC1 or CV_8UC3 in function ”错误提示。经研究发现2维ndarray元素数据类型必须转换成np.uint8类型才行。

import numpy as np
import cv2
a = np.ones((2,3))
b = a.astype(np.uint8)
img1 = cv2.applyColorMap(a, cv2.COLORMAP_JET)  # 报错
img2 = cv2.applyColorMap(b, cv2.COLORMAP_JET)  # 正常