使用方法:
tf.image.rgb_to_grayscale(images, name=None)
images 准备转换的 RGB 张量。最后一个维度必须为 3,必须包含 RGB 值
name 名称(可选)
代码:
#预处理函数
def preprocess_image(image):
image = tf.image.decode_jpeg(image, channels=3)
image = tf.image.rgb_to_grayscale(image)#RGB图像转为灰度图片
image = tf.image.resize(image, [100, 210])#图片太大,需要缩放图片尺寸,不然GPU无过多资源
return image/255.