当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python mxnet.image.imread用法及代码示例


用法:

mxnet.image.imread(filename, *args, **kwargs)

参数：

  • filename：(str) - 要加载的图像文件的名称。
  • flag：({0, 1}, default 1) - 1 用于三通道颜色输出。 0 表示灰度输出。
  • to_rgb：(bool, default True) - RGB 格式输出为真(MXNet 默认)。 BGR 格式输出为 False(OpenCV 默认)。
  • out：(NDArray, optional) - 输出缓冲区。采用None用于自动分配。

返回：

包含图像的NDArray。

返回类型：

ND阵列

读取图像并将其解码为 NDArray。

注意：

imread 使用 OpenCV(不是 CV2 Python 库)。 MXNet 必须使用 USE_OPENCV=1 构建，imdecode 才能工作。

示例：

>>> mx.img.imread("flower.jpg")
<NDArray 224x224x3 @cpu(0)>

设置flag参数为0得到灰度输出

>>> mx.img.imread("flower.jpg", flag=0)
<NDArray 224x224x1 @cpu(0)>

将 to_rgb 参数设置为 0 以获取 OpenCV 格式 (BGR) 的输出

>>> mx.img.imread("flower.jpg", to_rgb=0)
<NDArray 224x224x3 @cpu(0)>

相关用法


注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.image.imread。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。