把NumPy图像保存
时间: 2023-07-04 09:10:57 浏览: 37
要将NumPy数组中的图像保存为文件,可以使用Python中的OpenCV库或Matplotlib库。以下是使用这两个库的示例代码:
使用OpenCV:
```python
import cv2
import numpy as np
# 创建一个随机的 NumPy 数组表示一张图像
image = np.random.rand(256, 256) * 255
image = image.astype(np.uint8) # 转换为 8 位整数类型
# 保存图像
cv2.imwrite('image.jpg', image)
```
在上面的例子中,我们首先创建了一个随机的256x256的NumPy数组,表示一张灰度图像。然后,我们将图像的所有像素值乘以255,以将其转换为8位整数类型。最后,我们使用OpenCV的`imwrite`函数将图像保存为JPEG文件。
使用Matplotlib:
```python
import matplotlib.pyplot as plt
import numpy as np
# 创建一个随机的 NumPy 数组表示一张图像
image = np.random.rand(256, 256)
# 保存图像
plt.imsave('image.png', image, cmap='gray')
```
在上面的例子中,我们首先创建了一个随机的256x256的NumPy数组,表示一张灰度图像。然后,我们使用Matplotlib的`imsave`函数将图像保存为PNG文件,并指定了灰度图像的色彩映射为灰度(cmap='gray')。