直接将图像numpy数组转换为灰度数组而不保存图像

Question

我有一个 NumPy 数组 img_array 一个图像的维度 (h,w,3)，这是某个函数的结果。我想把这个 NumPy 数组直接转成灰度。

可能的解决方案： 使用cv2.imwrite(path) 将img_array 保存为图像。然后用cv2.imread(path, cv2.GRAYSCALE)再次阅读

但是，我正在寻找这样的东西：

def convert_array_to_grayscale_array(img_array):
        do something...
        return grayscare_version

我已经尝试过cv2.imread(img_array, CV2.GRAYSCALE)，但是抛出了img_array must be a file pathname 的错误。

我认为保存单独的图像会消耗更多的磁盘空间。有没有更好的方法可以使用或不使用 OpenCV 库函数。

Answer 1

scikit-image具有颜色转换功能：https://scikit-image.org/docs/dev/auto_examples/color_exposure/plot_rgb_to_gray.html

from skimage.color import rgb2gray

grayscale = rgb2gray(img_array)