在 Python 中使用 OpenCv2 编写多个映像

Question

我得到了一个充满图像的数据集。我的目标是将此图像转换为灰度。我可以很容易地用一个图像做到这一点，但我需要做多个图像。这是我的代码;

import cv2

import glob

path = "/path/*.jpg"

for file in glob.glob(path):

    img = cv2.imread(file)

    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

    cv2.imshow('Gray image', gray)

    cv2.imwrite('/path/cat' + '_gray.jpg', gray)

当我运行此脚本时，它只转换单个图像，但我的目标是转换路径中的所有图像。我真的是图像处理和OpenCv2模块的初学者。现在已经感谢您的帮助。

Answer 1

'/path/cat' + '_gray.jpg'始终为您提供 ，而不考虑输入文件名。/path/cat_gray.jpg您可能希望在此处使用原始文件名的某些部分，您可以使用 os.path 模块中的函数找到这些部分。例如：import cv2import globimport os.pathpath = "/path/*.jpg"for file in glob.glob(path):    img = cv2.imread(file)    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)    cv2.imshow('Gray image', gray)    basename = os.path.basename(file)  # e.g. MyPhoto.jpg    name = os.path.splitext(basename)[0]  # e.g. MyPhoto    cv2.imwrite('/path/' + name + '_gray.jpg', gray)