numpy.ndarray转图片并保存
时间: 2023-12-25 13:05:52 浏览: 163
Python 实现Image和Ndarray互相转换
可以使用PIL库将numpy.ndarray转换为图片并保存。以下是示例代码:
```python
import numpy as np
from PIL import Image
# 创建一个随机的 128x128 的 numpy 数组
arr = np.random.randint(0, 255, size=(128, 128, 3), dtype=np.uint8)
# 创建 Image 对象
img = Image.fromarray(arr)
# 保存图片
img.save('example.png')
```
其中,`arr` 是一个 128x128 的 numpy 数组,表示一张 RGB 图片。通过 `Image.fromarray()` 方法将其转换为 PIL 图像对象,然后通过 `img.save()` 方法保存为 PNG 格式的文件。
阅读全文