mindspore.dataset.vision.read_image
- mindspore.dataset.vision.read_image(filename, mode=ImageReadMode.UNCHANGED)[源代码]
读取图像文件并解码为3通道RGB彩色数据或灰度数据。 支持的文件类型有JPEG、PNG、BMP和TIFF。
- 参数:
filename (str) - 待读取图像文件路径。
mode (
ImageReadMode
, 可选) - 图像读取模式。它可以是ImageReadMode.UNCHANGED
、ImageReadMode.GRAYSCALE
、ImageReadMode.COLOR
。 默认值:ImageReadMode.UNCHANGED
。ImageReadMode.UNCHANGED - 按照图像原始格式读取。
ImageReadMode.GRAYSCALE - 读取并转为单通道灰度数据。
ImageReadMode.COLOR - 读取并换为3通道RGB彩色数据。
- 返回:
numpy.ndarray, 三维uint8类型数据,shape为(H, W, C)。
- 异常:
TypeError - 如果 filename 不是str类型。
TypeError - 如果 mode 不是
mindspore.dataset.vision.ImageReadMode
类型。RuntimeError - 如果 filename 不存在或不是普通文件或由于格式等原因无法正常读取。
- 支持平台:
CPU
样例:
>>> import mindspore.dataset.vision as vision >>> from mindspore.dataset.vision import ImageReadMode >>> output = vision.read_image("/path/to/image_file", ImageReadMode.UNCHANGED)