批量将png格式转为npy
时间: 2023-07-30 20:05:57 浏览: 324
可以使用Python中的numpy和Pillow库来实现将PNG格式转换为Numpy数组(即npy文件)的功能。以下是一个示例代码:
```python
import numpy as np
from PIL import Image
import os
# 定义PNG文件路径和Numpy文件保存路径
png_path = 'path/to/png/files/'
npy_path = 'path/to/npy/files/'
# 获取PNG文件列表
png_files = os.listdir(png_path)
# 遍历PNG文件列表,将每个PNG文件转换为Numpy数组并保存为npy文件
for png_file in png_files:
# 读取PNG文件
png = Image.open(os.path.join(png_path, png_file))
# 将PNG文件转换为Numpy数组
npy = np.array(png)
# 保存为npy文件
np.save(os.path.join(npy_path, os.path.splitext(png_file)[0] + '.npy'), npy)
```
上述代码会将指定路径下的所有PNG文件转换为Numpy数组,并保存为同名的npy文件。其中,我们使用了Pillow库中的`Image.open()`函数来读取PNG文件,使用numpy的`np.array()`函数将PNG文件转换为Numpy数组,最后使用numpy的`np.save()`函数将Numpy数组保存为npy文件。
阅读全文