Python OpenCV – 将图片保存到指定文件夹

在本文中，我们将介绍如何使用Python OpenCV库将图像保存到指定的文件夹中。OpenCV是一个强大的计算机视觉库，提供了许多图像处理和分析的功能。保存图像到特定文件夹可以帮助我们整理和管理图像数据。

1. 检查文件夹是否存在

在保存图像之前，我们首先需要检查指定的文件夹是否已经存在。如果文件夹不存在，我们可以使用os模块创建它。下面是一个示例代码：

import os

folder_path = "images"
if not os.path.exists(folder_path):
    os.makedirs(folder_path)

在上面的代码中，我们首先定义了一个变量folder_path，指定了要保存的文件夹路径。然后，使用os.path.exists()函数检查文件夹是否已经存在。如果文件夹不存在，我们使用os.makedirs()函数创建它。

2. 保存图像到指定文件夹

一旦文件夹存在，我们就可以使用OpenCV库保存图像了。OpenCV提供了cv2.imwrite()函数，可以将图像保存到指定的文件夹中。下面是保存图像的示例代码：

import cv2

image = cv2.imread("input_image.jpg")
save_path = "images/output_image.jpg"
cv2.imwrite(save_path, image)

在上面的代码中，我们首先使用cv2.imread()函数读取输入图像，并将其保存在image变量中。然后，我们定义了一个保存图像的路径save_path，该路径指定了保存图像的文件夹和文件名。最后，我们使用cv2.imwrite()函数将图像保存到指定的文件夹中。

3. 修改保存图像的文件名

有时候，我们可能希望修改保存图像的文件名，以便更好地描述图像内容或方便后续处理。我们可以使用os.path.join()函数将文件夹路径和新的文件名组合起来，以达到修改文件名的目的。下面是一个示例代码：

import cv2
import os

image = cv2.imread("input_image.jpg")
folder_path = "images"
file_name = "output_image.jpg"
save_path = os.path.join(folder_path, file_name)
cv2.imwrite(save_path, image)

在上面的代码中，我们通过定义一个变量file_name来指定新的图像文件名。然后，使用os.path.join()函数将文件夹路径和文件名组合起来，得到最终的保存路径。

4. 批量保存图像

有时候，我们可能需要批量保存一系列图像到指定的文件夹中。可以通过使用循环来实现。下面是一个示例代码：

import cv2
import os

folder_path = "images"
file_names = ["image1.jpg", "image2.jpg", "image3.jpg"]

for file_name in file_names:
    image_path = os.path.join(folder_path, file_name)
    image = cv2.imread(image_path)

    save_path = os.path.join(folder_path, "output_" + file_name)
    cv2.imwrite(save_path, image)

在上面的代码中，我们定义了一个文件名列表file_names，其中包含了要保存的图像文件名。然后，使用循环遍历文件名列表，并将每个图像保存到指定的文件夹中。通过给保存文件名添加前缀output_，我们可以在保存时对文件进行重命名。