基于python opencv 多进程处理图像

在Python中使用OpenCV进行图像处理时，由于图像数据通常较大，单进程处理可能会面临性能瓶颈。为了提升处理速度，我们可以利用多进程（multiprocessing）来并行处理图像。多进程可以充分利用多核CPU的计算资源，每个进程在独立的内存空间中运行，从而避免GIL（全局解释器锁）对性能的影响（尽管GIL主要影响多线程在CPython中的性能）。

下面是一个使用Python的multiprocessing库和OpenCV来并行处理图像的基本示例。我们将定义一个函数来处理图像（例如，将图像转换为灰度），然后使用multiprocessing.Pool来并行地处理一个图像列表。

步骤 1: 导入必要的库

import cv2
from multiprocessing import Pool
import numpy as np

步骤 2: 定义图像处理函数

该函数将作为多进程池中的工作单元。

def process_image(image_path):
    # 读取图像
    img = cv2.imread(image_path)
    if img is None:
        return None
    # 转换为灰度图像
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    # 可以在这里添加更多的图像处理步骤
    return gray

步骤 3: 准备图像列表

准备你想要处理的图像文件的路径列表。

image_paths = ['path/to/image1.jpg', 'path/to/image2.jpg', 'path/to/image3.jpg']

步骤 4: 使用多进程池处理图像

def main():
    with Pool(processes=4) as pool:  # 设定进程池中的进程数
        # map函数将image_paths列表中的每个元素作为参数传递给process_image函数，并返回结果列表
        results = pool.map(process_image, image_paths)
    
    # 遍历结果并显示或保存处理后的图像
    for result in results:
        if result is not None:
            cv2.imshow('Processed Image', result)
            cv2.waitKey(0)
            cv2.destroyAllWindows()
            # 如果需要，可以在这里保存图像
            # cv2.imwrite('processed_image.jpg', result)

if __name__ == '__main__':
    main()

注意：

• cv2.imshow 和 cv2.waitKey 在多进程环境中可能会遇到问题，因为它们是GUI操作且依赖于主线程。在实际应用中，你可能需要将处理后的图像数据发送回主进程进行显示或保存。
• 上述示例中，cv2.imshow 实际上只在最后一个图像上有效，因为窗口会立即关闭。为了查看所有图像，你可能需要调整代码来逐个显示它们，或者将所有处理后的图像保存到文件中。
• 当你处理大量图像或非常大的图像时，请确保你的系统有足够的内存来处理多个图像副本。
• 对于更复杂的图像处理任务，你可能需要考虑使用更高级的并行处理框架，如Dask或Ray，这些框架提供了更丰富的数据并行和任务调度的功能。

完整代码

from PIL import Image, ImageDraw, ImageFont
import datetime
import random
import os
import glob
from multiprocessing import Pool
random.seed(2024)

def process_image(image_full_path):
    # 如果需要保存图片到文件
    image.save(os.path.join(label_path,image_name))
if __name__ == '__main__':
    # 读取图片
    image_path = 'data/label/*.*'  # 图片路径
    image_list=glob.glob(image_path)

    with Pool(processes=16) as pool:  # 设定进程池中的进程数
        # map函数将image_paths列表中的每个元素作为参数传递给process_image函数，并返回结果列表
        results = pool.map(process_image, image_list)