将文件夹中的多张图片转为numpy数组
时间: 2023-05-20 12:01:56 浏览: 336
Img2Lcd(将图片转换为数组的工具)
5星 · 资源好评率100%
你可以使用Python中的Pillow库来将图片转换为numpy数组。以下是一个示例代码:
```python
from PIL import Image
import numpy as np
import os
# 定义一个函数,将图片转换为numpy数组
def image_to_array(image_path):
with Image.open(image_path) as img:
img_array = np.array(img)
return img_array
# 定义一个函数,将文件夹中的所有图片转换为numpy数组
def folder_to_array(folder_path):
img_list = os.listdir(folder_path)
img_arrays = []
for img_name in img_list:
img_path = os.path.join(folder_path, img_name)
img_array = image_to_array(img_path)
img_arrays.append(img_array)
return img_arrays
# 调用函数,将文件夹中的所有图片转换为numpy数组
folder_path = 'path/to/folder'
img_arrays = folder_to_array(folder_path)
```
你需要将`path/to/folder`替换为你实际的文件夹路径。这个代码将会读取文件夹中的所有图片,并将它们转换为numpy数组,存储在`img_arrays`列表中。
阅读全文