将一个已知的numpy.ndarray转换为图像保存
时间: 2024-03-10 18:49:12 浏览: 61
save-pixels:将 ndarray 保存为图像
要将一个已知的 `numpy.ndarray` 转换为图像并保存到文件,可以使用 `PIL` 库中的 `Image` 类和 `save()` 方法。具体步骤如下:
1. 导入 `PIL` 库和 `numpy` 库:
```python
from PIL import Image
import numpy as np
```
2. 将 `numpy.ndarray` 转换为 `PIL.Image` 对象:
```python
img = Image.fromarray(array)
```
其中,`array` 是你要转换的 `numpy.ndarray` 对象。
3. 可选:调整图像格式和尺寸:
```python
img = img.convert('RGB') # 将图像格式转换为 RGB
img = img.resize((width, height)) # 调整图像尺寸
```
其中,`width` 和 `height` 是你想要调整的图像宽度和高度。
4. 将图像保存为文件:
```python
img.save('output.png') # 将图像保存为文件
```
其中,`output.png` 是你想要保存的图像文件名。
完整代码示例:
```python
from PIL import Image
import numpy as np
# 加载一个已知的 numpy.ndarray 对象
array = np.load('image.npy')
# 将 numpy.ndarray 转换为 PIL.Image 对象
img = Image.fromarray(array)
# 调整图像格式和尺寸
img = img.convert('RGB')
img = img.resize((512, 512))
# 将图像保存为文件
img.save('output.png')
```
在这个例子中,我们首先加载了一个已知的 `numpy.ndarray` 对象,然后将其转换为 `PIL.Image` 对象并调整了格式和尺寸,最后将图像保存为文件。
阅读全文