Jupiter NotebookにMXNet NDArrayをイメージとして表示するには？

Question

イメージデータが入っているMXNet NDArrayがあります。 Jupiter Notebookで画像としてNDArrayをレンダリングするにはどうすればよいですか？ここで

type(data) 
mxnet.ndarray.ndarray.NDArray 

data.shape 
(3, 759, 1012) 

Answer 1

はそれを行う方法です：

1. はnumpyの配列にMXNet NDArrayに変換します。
2. チャネルを最後の次元に移動するために配列を転置します。
3. 配列をuint8（0〜255）に変換します。
4. matplotlibを使用して配列をレンダリングします。ここで

コードです：

import mxnet as mx 
import numpy as np 
from matplotlib import pyplot as plt 

def render_as_image(a): 
    img = a.asnumpy() # convert to numpy array 
    img = img.transpose((1, 2, 0)) # Move channel to the last dimension 
    img = img.astype(np.uint8) # use uint8 (0-255) 

    plt.imshow(img) 
    plt.show() 

あなたはその後、render_as_imageを呼び出すことで、配列をレンダリングすることができます。

render_as_image(data)