cv2.cvtColor()将二维转化为彩色图像

我们如果要将一维的图像转化为三维的彩色图像

import cv2
img = cv2.cvtColor(img.squeeze(0), cv2.COLOR_GRAY2BGR) 

这里的img.squeeze为二维数据，img为三维数据，所以压缩掉一个维度 。这样就将图像转化为了三维彩色图像。

cv2.imshow('Image', img)
cv2.waitKey(0)
cv2.destroyAllWindows()

将图像进行可视化。