问题:
我正尝试使用matplotlib
读取RGB图像并将其转换为灰度。
在matlab中,我使用这个:
1
|
img = rgb2gray(imread( 'image.png' ));
|
在matplotlib tutorial中他们没有覆盖它。他们只是在图像中阅读
1
2
|
import matplotlib.image as mpimg
img = mpimg.imread( 'image.png' )
|
然后他们切片数组,但是这不是从我所了解的将RGB转换为灰度。
1
|
lum_img = img[:,:, 0 ]
|
编辑:
我发现很难相信numpy或matplotlib没有内置函数来从rgb转换为灰色。这不是图像处理中的常见操作吗?
我写了一个非常简单的函数,它可以在5分钟内使用imread
导入的图像。这是非常低效的,但这就是为什么我希望内置专业实施。
塞巴斯蒂安改善了我的功能,但我仍然希望找到内置的一个。
matlab的(NTSC / PAL)实现:
1
2
3
4
5
6
7
8
|
import numpy as np
def rgb2gray(rgb):
r, g, b = rgb[:,:, 0 ], rgb[:,:, 1 ], rgb[:,:, 2 ]
gray = 0.2989 * r + 0.5870 * g + 0.1140 * b
return gray
|
回答:
如何使用PIL
1
2
3
|
from PIL import Image
img = Image. open ( 'image.png' ).convert( 'LA' )
img.save( 'greyscale.png' )
|
使用matplotlib和the formula
1
|
Y' = 0.299 R + 0.587 G + 0.114 B
|
你可以这样做:
1
2
3
4
5
6
7
8
9
10
11
|
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
def rgb2gray(rgb):
return np.dot(rgb[...,: 3 ], [ 0.299 , 0.587 , 0.114 ])
img = mpimg.imread( 'image.png' )
gray = rgb2gray(img)
plt.imshow(gray, cmap = plt.get_cmap( 'gray' ))
plt.show() |