python - 有没有一种方法可以将图像保存在 OpenCV 执行期间创建的子文件夹中

Question

这是我第一次尝试使用 OpenCV 制作 python 项目。我想知道是否有办法将 OpenCV 捕获的图像存储到子文件夹中。

目前运行ImageCapture功能时，它会为所有用户拍摄照片并将其存储在同一个文件夹中。

for (x,y,w,h) in faces:
     cv2.rectangle(frame,(x,y),(x+w,y+h),(255,0,0),2)        
     sampleNum=sampleNum+1
     cv2.imwrite("TrainingImage\ "+name +"."+Id +'.'+ str(sampleNum) + ".jpg", gray[y:y+h,x:x+w])

名称和 ID 是在调用 ImageCapture 函数之前由用户提供的。是否有任何方法可以将图像存储到具有用户提供的名称的子文件夹中？

Answer 1

创建文件夹不是 OpenCV 的工作，您可以使用您使用的编程语言来完成它。

就您而言，我想这就是您要找的。

import os, errno
import cv2

for (x,y,w,h) in faces:
    cv2.rectangle(frame,(x,y),(x+w,y+h),(255,0,0),2)        
    sampleNum=sampleNum+1
    try:
        if not os.path.exists("TrainingImage/" + name):
            os.makedirs("TrainingImage/" + name)
        cv2.imwrite("TrainingImage/"+ name +"/" + name + "." + str(Id) +'.'+ str(sampleNum) + ".jpg", gray[y:y+h,x:x+w])
    except OSError as e:
        if e.errno != errno.EEXIST:
            raise