python 颜画像切り出しスクリプト

Posted 2021-05-09

import cv2
import sys
import os
import os.path

def detect(filename, cascade_file = "../lbpcascade_animeface.xml"):
    if not os.path.isfile(cascade_file):
        raise RuntimeError("%s: not found" % cascade_file)

    cascade = cv2.CascadeClassifier(cascade_file)
    image = cv2.imread(filename)
    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    gray = cv2.equalizeHist(gray)

    faces = cascade.detectMultiScale(gray,
                                     # detector options
                                     scaleFactor = 1.05,
                                     minNeighbors = 5,
                                     minSize = (32, 32))
    for (x, y, w, h) in faces:
        cv2.imwrite("./out/" + filename, image[y:y+h, x:x+w])

# main
files = os.listdir("./")

for filepath in sorted(files):
    if filepath.endswith(".png"):
        print(filepath)
        detect(filepath)