обрезать лицо из видео и сохранить как изображение - PullRequest
0 голосов
/ 06 апреля 2020

Я пытаюсь обрезать лицо каждый раз, когда оно появляется в видео, но создается впечатление, что оно не создает файл изображения.

Я включил строку imshow, чтобы увидеть, есть ли изображение, создаваемое из ограничивающих рамок, и есть ли лицо, которое обрезается. Любая помощь будет оценена.

Код:

import numpy as np

import cv2

import os



detector= cv2.CascadeClassifier("haarcascade_frontalface_default.xml")

cap = cv2.VideoCapture("smallvid.mp4")    

while(True):

    ret, img = cap.read()

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

    faces = detector.detectMultiScale(gray, 1.3, 5)

    count = 1 
    for (x,y,w,h) in faces:
        cropped = img[ y : y+h, x : x+w ]
        cv2.imwrite("cropped_face" + str(id) + ".png", cropped)
        count=count+1
        cv2.imshow(out_dir+str(count), cropped); # show an image of each face
        cv2.rectangle(img,(x,y),(x+w,y+h),(255,0,0),2) # create bounding box around face


    cv2.imshow('frame',img)

    if cv2.waitKey(1) & 0xFF == ord('q'):

        break

cap.release()

cv2.destroyAllWindows()
enter code here

1 Ответ

0 голосов
/ 06 апреля 2020

Вы можете обрезать, как это.

img[y:y+w, x:x+w]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...