pythonopencvimage-processingface-recognition

OpenCV Python face recognition only one face recognition


Hi,Everyone We are working on a face recognition application when ı run the code below, as you can see in the photo it works only in one face(red square) it doesnt scan other faces in the traning-data ı guess my predict function only run one time. dont in the loop.

Processed Image : LINK

 # coding: utf-8
 import cv2
 import os
 import numpy as np
 suclular = ["Bilinmeyen", "Veli Eroglu", "Ali Eroglu"]


 def detect_face(img):
     # ALGORİMA için Gri Yapıyoruz.
     gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
     # yüz tanımlama için geereken haarcascade
     face_cascade = cv2.CascadeClassifier(
         'opencv-files/lbpcascade_frontalface.xml')
     faces = face_cascade.detectMultiScale(
         gray, scaleFactor=1.2, minNeighbors=5)  # YÜZ TANIMLAMA
     for (x, y, w, h) in faces:
         img = cv2.rectangle(img, (x, y), (x + w, y + h), (255, 0, 0), 2)
     if (len(faces) == 0):
         return None, None  # Yuz bulunamazsa...
     (x, y, w, h) = faces[0]
     return gray[y:y + w, x:x + h], faces[0]

 face_recognizer = cv2.face.LBPHFaceRecognizer_create()
 face_recognizer.train(faces, np.array(labels))


 def draw_rectangle(img, rect):
     (x, y, w, h) = rect
     cv2.rectangle(img, (x, y), (x + w, y + h), (0, 0, 255), 2)


 def draw_text(img, text, x, y):
     cv2.putText(img, text, (x, y), cv2.FONT_HERSHEY_PLAIN, 1.5, (0, 0, 255), 2)


 def predict(test_img):
     img = test_img.copy()
     face, rect = detect_face(img)
     label, confidence = face_recognizer.predict(face)
     print(confidence)
     label_text = suclular[label]
     if confidence > 42 and confidence < 70:
         label_text = "Tespit Edilemedi."
         print(label_text)
     elif confidence > 70:
         label_text = "Bilinmiyor"
     draw_rectangle(img, rect)
     draw_text(img, label_text, rect[0], rect[1] - 5)

     return img


 print("Predicting images...")
 test_img1 = cv2.imread("test-data/test8jpg.jpg")
 predicted_img1 = predict(test_img1)
 print("Prediction complete")
 cv2.imshow("SONUC", cv2.resize(predicted_img1, (400, 500)))
 cv2.waitKey(0)
 cv2.destroyAllWindows()
 cv2.waitKey(1)
 cv2.destroyAllWindows()

Solution

  • You prediction should be with in the for loop... You are only returning one face from your detect_face function which is the last face even though you are looping through each face and making a rectangle for each face... You should do something like this:

    def predict_face(img):
         # ALGORİMA için Gri Yapıyoruz.
         gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
         # yüz tanımlama için geereken haarcascade
         face_cascade = cv2.CascadeClassifier(
             'opencv-files/lbpcascade_frontalface.xml')
         faces = face_cascade.detectMultiScale(
             gray, scaleFactor=1.2, minNeighbors=5)  # YÜZ TANIMLAMA
         detected_faces = []
         i = 0
         for (x, y, w, h) in faces:
             img = cv2.rectangle(img, (x, y), (x + w, y + h), (255, 0, 0), 2)
             detected_face = gray[y:y + w, x:x + h]
             label, confidence = face_recognizer.predict(detected_face)  # Prediction inside for loop
             draw_rectangle(img, faces[i])    # draw the red rectangles for every predicted face
             draw_text(img, label, x, y - 5)  # draw the predicted label on top of the box
             i += 1