python - 我想使用Python将32 * 32 * 3 RGB图像转换为32 * 32 * 1维度的灰度图像

Question

以下是我将 RGB 图像转换为灰度图像的函数。

我的输入图像是32*32*3，而输出尺寸看起来像32*32，但我正在寻找32*32*1 。我需要调整该图像的大小或比例吗？

有什么想法吗？

def rgb2gray(rgb):
    return np.dot(rgb[...,:3], [0.299, 0.587, 0.114])

Answer 1

如果您希望在 Tensorflow 图本身中进行转换，您可以使用此函数:https://www.tensorflow.org/api_docs/python/tf/image/rgb_to_grayscale

tf.image.rgb_to_grayscale(input_images)

另外，看起来您正在回答自己的问题。有什么问题

def rgb2gray(rgb):
  return np.dot(rgb[...,:3], [0.299, 0.587, 0.114])

祝你好运!