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


Python mxnet.recordio.unpack_img用法及代碼示例

用法:

mxnet.recordio.unpack_img(s, iscolor=-1)

參數

  • s(str) - 來自的字符串緩衝區MXRecordIO.read.
  • iscolor(int) - 圖像格式選項cv2.imdecode.

返回

  • header(IRHeader) - 圖像記錄的標題。
  • img(numpy.ndarray) - 解壓後的圖像。

將 MXImageRecord 解壓縮為圖像。

例子

>>> record = mx.recordio.MXRecordIO('test.rec', 'r')
>>> item = record.read()
>>> header, img = mx.recordio.unpack_img(item)
>>> header
HEADER(flag=0, label=14.0, id=20129312, id2=0)
>>> img
array([[[ 23,  27,  45],
        [ 28,  32,  50],
        ...,
        [ 36,  40,  59],
        [ 35,  39,  58]],
       ...,
       [[ 91,  92, 113],
        [ 97,  98, 119],
        ...,
        [168, 169, 167],
        [166, 167, 165]]], dtype=uint8)

相關用法


注:本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.recordio.unpack_img。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。