如何在Jupyter Notebook中将MXNet NDArray显示为图像？

作者: 甲基蓝
发布时间: 2024-04-09 11:52:38 (7天前)
转自：

2 条回复

0#
回复此人
猫南北 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 这是怎么做的： </p> <OL> <LI> 将MXNet NDArray转换为numpy数组。 </LI> <LI> 转置数组以将通道移动到最后一个维度。 </LI> <LI> 将数组转换为uint8（0到255）。 </LI> <LI> 使用matplotlib渲染数组。 </LI> </醇> <P> 这是代码： </p> <pre> <code> 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() </code> </pre> <P> 然后，您可以通过调用呈现数组 <code> render_as_image </code> 。 </p> <pre> <code> render_as_image(data) </code> </pre> </DIV>

编辑

登录后才能参与评论