skimage.color模块小坑：rgb2gray

原创

于 2024-07-20 15:38:00 发布 · 590 阅读

12 ·

CC 4.0 BY-SA版权

文章标签：

#python

文章目录

skimage.color模块小坑：rgb2gray
- 问题记录
- 改好的代码

skimage.color模块小坑：rgb2gray

问题记录

skimage.color模块中的rgb2gray可以将彩色图像变成灰度图。但是我今天的代码缺除了错误，读入彩色图输出灰度图像全是 0。问题代码如下

from PIL import Image
from skimage import color
import numpy as np

def path_to_gray(path: str) -> Image.Image:
    """
    Load an image from the given path and convert it to Gray format.
    """
    image = Image.open(path)
    if image.mode == "RGB":
        image = color.rgb2gray(image)
    elif image.mode != 'L':
        raise ValueError(f"Only RGB and Gray images are supported, not {
     image.mode}")
    return Image.fromarray(np.array(image).astype(np.uint8)