python 批量读取图像 np

原创

mob649e815375e5 2023-11-10 10:29:20 ©著作权

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

Python批量读取图像

Python是一种强大的编程语言，它提供了丰富的库和模块，使得处理图像变得非常简单。在本文中，我们将学习如何使用Python批量读取图像，并进行相应的处理。

准备工作

在开始之前，我们需要安装一些必要的库。其中，numpy是一个用于处理多维数组的库，opencv-python是OpenCV的Python接口，用于图像处理。可以使用以下命令来安装这些库：

!pip install numpy
!pip install opencv-python

代码示例

下面是一个使用Python批量读取图像的示例代码：

import cv2
import os
import numpy as np

# 图像文件夹路径
image_folder = "images/"

# 读取文件夹中的图像文件名
image_files = os.listdir(image_folder)

# 创建一个空的Numpy数组，用于存储图像数据
image_data = np.empty((len(image_files),))

# 遍历图像文件名
for i, image_file in enumerate(image_files):
    # 拼接图像文件路径
    image_path = os.path.join(image_folder, image_file)
    
    # 读取图像文件
    image = cv2.imread(image_path)
    
    # 将图像数据存储到Numpy数组中
    image_data[i] = image

# 打印图像数据
print(image_data)

以上代码中，我们首先导入了需要的库，然后指定了图像文件夹的路径。接下来，使用os.listdir函数读取文件夹中的图像文件名，并使用enumerate函数遍历这些文件名。

在循环中，我们通过拼接图像文件路径，并使用cv2.imread函数读取图像文件。然后，将图像数据存储到先前创建的Numpy数组中。

最后，我们打印图像数据，以验证是否成功读取了图像。

序列图

下面是一个使用mermaid语法绘制的序列图，展示了整个图像读取的过程：

sequenceDiagram
    participant Python
    participant OpenCV
    participant OS
    
    Python->>OS: 读取文件夹中的图像文件名
    OS-->>Python: 图像文件名列表
    loop 遍历图像文件名
        Python->>OS: 拼接图像文件路径
        OS-->>Python: 图像文件路径
        Python->>OpenCV: 读取图像文件
        OpenCV-->>Python: 图像数据
        Python->>Python: 存储图像数据到数组
    end
    Python-->>Python: 打印图像数据

Gantt图

下面是一个使用mermaid语法绘制的甘特图，展示了图像读取的时间分布：

gantt
    dateFormat  YYYY-MM-DD
    title 图像读取时间分布
    section 读取图像
    读取图像文件名    :done, 2021-01-01, 1d
    拼接图像文件路径     :done, 2021-01-02, 1d
    读取图像文件     :done, 2021-01-03, 1d
    存储图像数据到数组     :done, 2021-01-04, 1d
    打印图像数据     :done, 2021-01-05, 1d