将numpy数组转化为图像数据集并保存到指定文件夹中
时间: 2023-05-18 12:03:34 浏览: 247
将图像保存成一定格式的数据文件
可以使用PIL库将numpy数组转化为图像数据集,并使用os库将其保存到指定文件夹中。以下是示例代码:
```python
import numpy as np
from PIL import Image
import os
# 创建一个3x3的numpy数组
arr = np.array([[255, 0, 0], [0, 255, 0], [0, 0, 255]], dtype=np.uint8)
# 将numpy数组转化为PIL图像对象
img = Image.fromarray(arr)
# 保存图像到指定文件夹中
if not os.path.exists('images'):
os.makedirs('images')
img.save('images/test.png')
```
这段代码将创建一个3x3的numpy数组,将其转化为PIL图像对象,并将其保存为PNG格式的图像文件到名为“images”的文件夹中。
阅读全文