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


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


用法:

mxnet.image.imdecode(buf, *args, **kwargs)

参数

  • buf(str/bytes/bytearray or numpy.ndarray) - 二进制图像数据作为字符串或 numpy ndarray。
  • flag(int, optional, default=1) - 1 用于三通道颜色输出。 0 表示灰度输出。
  • to_rgb(int, optional, default=1) - 1 用于 RGB 格式输出(MXNet 默认)。 0 用于 BGR 格式输出(OpenCV 默认)。
  • out(NDArray, optional) - 输出缓冲区。采用None用于自动分配。

返回

包含图像的NDArray

返回类型

ND阵列

将图像解码为 NDArray。

注意

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

示例

>>> with open("flower.jpg", 'rb') as fp:
...     str_image = fp.read()
...
>>> image = mx.img.imdecode(str_image)
>>> image
<NDArray 224x224x3 @cpu(0)>

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

>>> with open("flower.jpg", 'rb') as fp:
...     str_image = fp.read()
...
>>> image = mx.img.imdecode(str_image, flag=0)
>>> image
<NDArray 224x224x1 @cpu(0)>

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

>>> with open("flower.jpg", 'rb') as fp:
...     str_image = fp.read()
...
>>> image = mx.img.imdecode(str_image, to_rgb=0)
>>> image
<NDArray 224x224x3 @cpu(0)>

相关用法


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