关于使用cv2.ColorMap的错误：error: (-5:Bad argument) cv::ColorMap only supports source images of type CV_8UC

error: (-5:Bad argument) cv::ColorMap only supports source images of type CV_8UC

当我们通过模型得到一个（3,w,h）的Tensor时，我们想使用cv.ColorMap变化一下图像相当于得到一个heatmap

1.报错代码：

att是一个(b,c,h,w)的shape
#1.去掉维度b
att_ = att.cpu().numpy()[0]
#2.将所有值缩放到【0，255】
att_ = np.clip(att_, 0, 1)*255
#3.使用cv2.applyColorMap转换
heat_att= np.transpose(cv2.applyColorMap(att_.astype(np.uint8), cv2.COLORMAP_JET), (2, 0, 1))

错误分析：因为applyColorMap需要的是一个二维的图像，但是我们传入的是一个三维的就不行

2.方法1：直接去掉维度c

att_ = att.cpu().numpy()[0][1]
att_ = np.clip(att_, 0, 1)*255
heat_att= np.transpose(cv2.applyColorMap(att_.astype(np.uint8), cv2.COLORMAP_JET), (2, 0, 1))

2. 方法2:使用cvtColor变换为灰度图

可以将att_使用

att_ = cv2.cvtColor(att_,cv2.COLOR_RGB2GRAY)