Python : Dynamically saving unknown faces in face recognition not working

Question

I've a script which runs on image and recognizes faces and return a list like :

[('Mike', (142, 464, 365, 241)),('Garry', (42, 364, 65, 141)),('unknown', (242, 564, 465, 341))]

The second tuple is the bounding box of the faces recognized. I've another script which uses webcam, recognizes faces in frames and shows them in the video feed. I want to automatically save "unknown" labelled faces in each frames, whenever it appears. My code :

from stat_face_recog import runonimage
video_capture = cv2.VideoCapture(0)
while True:
    ret, frame = video_capture.read()
    cv2.imwrite("new.png",frame)
    final_pred = runonimage(img_path = "new.png")
    read_img = cv2.imread("new.png")
    for name, (top, right, bottom, left) in final_pred:        
        cv2.rectangle(frame, (left, top), (right, bottom), (0, 0, 255), 2)
        cv2.rectangle(frame, (left, bottom - 35), (right, bottom), (0, 0, 255), cv2.FILLED)
        font = cv2.FONT_HERSHEY_DUPLEX
        cv2.putText(frame, name, (left + 6, bottom - 6), font, 1.0, (255, 255, 255), 1)
        unknown_counter = 0
        if name == "unknown":
            unknowns_name = "unknown" + str(unknown_counter) + ".png"
            (new_top, new_right, new_bottom, new_left) = (int(0.8 * top), int(1.2* right), int(1.2*bottom), int(0.8*left))
            cv2.imwrite(unknowns_name,read_img[new_top:new_bottom, new_left:new_right])
            unknown_counter += 1
    cv2.imshow('Video', frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
video_capture.release()
cv2.destroyAllWindows()

But the problem is while it is recognizing untrained people pictures unknown, it's not saving those unknown faces. Only one image named "unknonw0.png" is saved every time. What is wrong in my code?

I don't know that much about python, but from a quick glance, you always set the unknown_counter to 0, everytime it loops. Put your counter out of the loop. — EInherjar
– EInherjar, Commented Sep 28, 2018 at 4:50

Chin Leung · Accepted Answer · 2018-09-29 22:36:03Z

2

You are resetting the unknown_counter to zero in your for loop. Therefore, the image is overwriting every time. Just move it outside the loops.

edited Sep 29, 2018 at 22:36

answered Sep 28, 2018 at 4:50

Chin Leung

15k4 gold badges38 silver badges64 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Python : Dynamically saving unknown faces in face recognition not working

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related