将批量NPZ文件生成图片,并将图片灰度化
时间: 2024-01-08 22:02:11 浏览: 128
对大量图片进行灰度处理
5星 · 资源好评率100%
可以使用类似的方法,将批量的npz文件中的数组数据生成灰度化图片。以下是一个示例代码:
```python
import numpy as np
from PIL import Image
import os
# 遍历文件夹中的所有npz文件
for file_name in os.listdir('npz_folder'):
if file_name.endswith('.npz'):
# 从npz文件中加载数据
data = np.load(os.path.join('npz_folder', file_name))
arr = data['arr_0']
# 将数组转换为灰度图像
img = Image.fromarray(arr.astype('uint8'), mode='L')
# 显示图像或保存为文件
img.show()
img.save(os.path.join('output_folder', file_name[:-4] + '.png'))
```
这里使用了`os`库来遍历文件夹中的所有npz文件。对于每个文件,加载数据后使用PIL库将其转换为灰度图像,并可以选择显示图像或将其保存为文件。注意,灰度图像的模式为`L`。最后将输出的图片保存到`output_folder`文件夹中,文件名与原npz文件一致但后缀为`.png`。
阅读全文