python中rgb2gray函数

在Python中，可以使用Pillow库中的Image模块来实现RGB图像转灰度图像的功能。具体代码如下：

from PIL import Image

def rgb2gray(rgb_img):
    # 将彩色图像转为灰度图像
    gray_img = rgb_img.convert('L')
    return gray_img

其中，参数`rgb_img`是一个Pillow库中的Image对象，表示原始的彩色图像；函数返回值`gray_img`也是一个Image对象，表示转换后的灰度图像。

在函数内部，通过调用`convert()`方法并传入参数`'L'`，即可将彩色图像转为灰度图像。这个参数表示将图像转换为灰度模式，即每个像素用一个8位的灰度值表示，取值范围为0~255，表示黑色到白色的不同程度。

实际使用时，可以先使用Pillow库中的`Image.open()`函数读取原始图像，然后将其作为参数传给`rgb2gray()`函数，得到转换后的灰度图像。示例代码如下：

from PIL import Image

def rgb2gray(rgb_img):
    # 将彩色图像转为灰度图像
    gray_img = rgb_img.convert('L')
    return gray_img

# 读取原始彩色图像
rgb_img = Image.open('lena.png')

# 转换为灰度图像
gray_img = rgb2gray(rgb_img)

# 保存灰度图像
gray_img.save('lena_gray.png')

在这个示例中，我们将`lena.png`这个彩色图像文件读入，然后调用`rgb2gray()`函数将其转换为灰度图像。最后将转换后的灰度图像保存为`lena_gray.png`文件。