将一个文件夹的图像放进数组
时间: 2024-04-11 17:28:20 浏览: 120
您可以使用Python的PIL库(Python Imaging Library)来读取文件夹中的图像,并将它们放进一个数组中。下面是一个示例代码:
```python
from PIL import Image
import os
def load_images_from_folder(folder):
images = []
for filename in os.listdir(folder):
img = Image.open(os.path.join(folder, filename))
if img is not None:
images.append(img)
return images
folder_path = 'your_folder_path' # 替换为您的文件夹路径
image_array = load_images_from_folder(folder_path)
```
这段代码首先导入了PIL库和os库。然后定义了一个函数`load_images_from_folder`,该函数接受一个文件夹路径作为输入,并返回一个包含所有图像的数组。在函数中,通过使用`os.listdir`函数列出文件夹中的所有文件名,并使用`Image.open`函数打开每个图像文件。如果图像文件不为空,则将其添加到`images`数组中。
最后,您可以将文件夹路径替换为您想要读取图像的实际文件夹路径,并将结果保存在`image_array`变量中。请注意,`image_array`将包含所有图像的PIL图像对象。如果需要,您可以进一步处理这些图像对象,比如转换为NumPy数组等。
阅读全文