python - 如何自动选择和处理/root/facedetect 目录中的每个图像而不是选择特定图像

Question

如果可以的话，有什么方法可以自动将所有检测到人脸的图像发送到一个单独的目录中，或者将它们压缩？

import numpy as np
import cv2

face_cascade = cv2.CascadeClassifier('/root/opencv/modules/objdetect/src/haarcascade_frontalface_alt.xml')

img = cv2.imread('/root/facedetect/1.jpg')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

faces = face_cascade.detectMultiScale(gray, 1.3, 5)
for (x,y,w,h) in faces:
    cv2.rectangle(img,(x,y),(x+w,y+h),(255,0,0),2)
    roi_gray = gray[y:y+h, x:x+w]
    roi_color = img[y:y+h, x:x+w]
cv2.imshow('img',img)
cv2.waitKey(0)
cv2.destroyAllWindows()

Answer 1

您可以使用 os模块获取文件夹中的所有图像。
然后使用 shutil模块将一些图像复制到另一个文件夹中。

    import osimage_path
    from shutil import copyfile

    images_folder = '/root/facedetect'
    images = os.listdir(images_folder)

    detected_faces_folder = '/root/detected_faces'

    for image in images:
        image_path = os.path.join(images_folder, image)
        img = cv2.imread(image_path)
        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
        ...
        ...

        if detected_faces:
            copyfile(image_path, os.path.join(detected_faces_folder, image))