Python opencv 读取文件夹中所有图像

原创

mob649e815c000a 2024-11-02 06:43:09 ©著作权

文章标签 OpenCV 读取图像状态图 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e815c000a的原创作品，请联系作者获取转载授权，否则将追究法律责任

使用Python OpenCV读取文件夹中的所有图像

在计算机视觉和图像处理领域，OpenCV（Open Source Computer Vision Library）是一个非常受欢迎的工具库。它提供了丰富的功能，包括图像读取、处理、特征提取等。本文将介绍如何使用Python中的OpenCV库来读取指定文件夹中的所有图像。我们将通过代码示例和状态图来帮助理解。

安装OpenCV

要开始使用OpenCV，我们需要首先确保已经安装了它。打开命令行工具并运行以下命令：

pip install opencv-python

一旦安装完成，我们就可以开始编写代码来读取图像了。

读取文件夹中的图像

假设我们有一个文件夹，里面存放了很多图像文件，我们希望一次性读取这些图像并显示它们。以下是一个简单的实现步骤：

导入必要的库。
获取文件夹中所有图像文件的路径。
使用OpenCV读取并显示每张图像。

下面是具体的代码示例：

import cv2
import os

# 定义一个函数来读取图像
def read_images_from_folder(folder_path):
    # 创建一个空列表来存储图像
    images = []
    
    # 遍历指定文件夹中的所有文件
    for filename in os.listdir(folder_path):
        if filename.endswith(('.png', '.jpg', '.jpeg', '.bmp', '.gif')):  # 只处理图像文件
            img_path = os.path.join(folder_path, filename)
            img = cv2.imread(img_path)  # 读取图像
            
            if img is not None:
                images.append(img)  # 将读取的图像添加到列表中
            else:
                print(f"无法读取图像文件: {img_path}")
    
    return images

# 使用示例
folder_path = 'path/to/your/image/folder'  # 替换为图像文件夹的路径
images = read_images_from_folder(folder_path)

# 显示图像
for i, image in enumerate(images):
    cv2.imshow(f'Image {i+1}', image)
    cv2.waitKey(0)  # 等待键盘输入
cv2.destroyAllWindows()  # 关闭所有图像窗口

代码解析

导入库：我们导入了 cv2 和 os 库，前者用于图像处理，后者用于文件操作。
定义函数：函数 read_images_from_folder 接受一个文件夹路径，然后遍历该文件夹下的所有文件。通过 os.listdir() 方法获取文件夹中的所有文件名。
图像过滤：在读取文件时，我们通过 endswith() 方法过滤出常见的图像文件格式（.png、.jpg、.jpeg、.bmp、.gif），以确保只处理所需的图像文件。
图像读取：使用 cv2.imread() 方法读取图像，并将其添加到一个列表中。如果图像读取失败，则输出错误信息。
显示图像：使用 cv2.imshow() 方法逐一显示读取的图像，并等待键盘输入以继续显示下一个图像。最后，调用 cv2.destroyAllWindows() 关闭所有图像窗口。

状态图

为了更清楚地展示整个处理流程，我们可以使用状态图，其中每个状态表示处理过程的一部分。以下是状态图的代码：

stateDiagram
    [*] --> Start
    Start --> Read_Folder
    Read_Folder --> Filter_Images
    Filter_Images --> Read_Images
    Read_Images --> Add_To_List
    Add_To_List --> Display_Images
    Display_Images --> [*]

在这个状态图中：