将numpy数组转为多张图像并保存到一个文件夹中
时间: 2023-05-18 21:06:11 浏览: 132
批量处理图片并保存至另一个文件夹中
5星 · 资源好评率100%
可以使用PIL库来实现将numpy数组转为多张图像并保存到一个文件夹中的功能。具体实现代码如下:
```python
from PIL import Image
import numpy as np
import os
def save_images_from_array(arr, folder_path):
if not os.path.exists(folder_path):
os.makedirs(folder_path)
for i in range(arr.shape[0]):
img = Image.fromarray(arr[i])
img.save(os.path.join(folder_path, f"{i}.png"))
# 示例代码
arr = np.random.randint(0, 255, size=(10, 28, 28))
save_images_from_array(arr, "images")
```
这段代码将生成一个10张28x28的随机图像,并保存到名为"images"的文件夹中。
阅读全文