opencv - 如何调整cvtColor灰度功能，使其保持RGB尺寸

Question

函数:cv2.cvtColor(img，cv2.COLOR_BGR2GRAY)

如果我输入32x32x3的图片(R / G / B为3)，它将输出32x32的图片。但是，我需要它返回一个32x32x1的图像，以便可以将其用于 tensorflow 。如何调整功能？

Answer 1

在TensorFlow中，您可以使用 tf.expand_dims(img, 2) 操作或TensorFlow索引运算符将32x32矩阵转换为32x32x1张量:

img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
img_tensor = tf.constant(img)
img_expanded = tf.expand_dims(img, 2)  # or `img[:, :, tf.newaxis]`

使用NumPy索引可以实现相同的目的:

import numpy as np
# ...
img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
img_expanded = img[:, :, np.newaxis]