Numpy:数据转换成image并保存

2023-12-13 22:50:16

Matplotlib库

若没有matplotlib库,则使用一下命令进行安装:

pip install matplotlib

conda install matplotlib

? ? ? 要将?float64?类型的 NumPy 数据转换为图像并保存,可以使用?matplotlib?库。以下是使用?matplotlib?的示例代码:

import numpy as np  
import matplotlib.pyplot as plt  
from PIL import Image  
  
# 创建一个随机的 float64 类型的 NumPy 数组  
data = np.random.rand(100, 100).astype(np.float64)  
  
# 将数据转换为图像  
img = plt.imshow(data, cmap='gray')  
  
# 保存图像为 PNG 文件  
plt.savefig('output.png')

? ? ? 上述代码将创建一个随机的?float64?类型的 NumPy 数组,并使用?matplotlib?的?imshow?函数将其转换为灰度图像。然后,使用?savefig?函数将图像保存为 PNG 文件。

PIL库

如果没有PLI库,可以使用以下命令进行安装:

pip install Pillow

conda install Pillow

使用?PIL?来保存图像,你可以使用以下代码:

from PIL import Image  
import numpy as np  
  
# 创建一个随机的 float64 类型的 NumPy 数组  
data = np.random.rand(100, 100).astype(np.float64)  
  
# 将数据转换为图像  
img = Image.fromarray((data * 255).astype(np.uint8))  
  
# 保存图像为 PNG 文件  
img.save('output.png')

? ? ? 这段代码与之前的示例类似,但使用?PIL?的?Image.fromarray?方法将 NumPy 数组转换为图像。然后,使用?save?方法将图像保存为 PNG 文件。注意,在将?float64?数据转换为图像之前,需要将其乘以 255 并转换为?uint8?类型。

文章来源:https://blog.csdn.net/Ethan_Rich/article/details/134947889
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。