批量实现 cv2.cvtColor(images, cv2.COLOR_GRAY2RGB)

  • 背景:我想在一批灰度图像上画彩色的框,那么首先需要对灰度图转为 RGB,但是 RGB 信息已经损失了,所以只能对通道复制 3 份,变成彩色的三通道,这里使用 OpenCV 的 cv2.cvtColor(images, cv2.COLOR_GRAY2RGB) 方法只能针对单张图片,所以本文给出能直接处理 batch 图像的 灰度转彩色
# init dummy mat
mat = np.zeros((32,28,28))
mat = np.expand_dims(mat, -1)
np.tile(mat, (1,1,1,3))

上面的代码实现了从 (32, 28, 28) -> (32, 28, 28, 3) 的转变

  • 解析:这里使用的 np.tile() 函数,这个函数很好理解,你有 n 个维度,那么你就需要对指定对相应维度复制的次数即可,上面的程序中,我先对 batch 在最后(-1) expand_dims 了一下,得到了(32,28,28,1),然后(1,1,1,3)是对相应的维度复制的次数

你可能感兴趣的:(OpenCV)