當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。