Распознавание лиц: показывать только одно лицо одновременно

Question

Я разрабатывал приложение для обнаружения лиц, интегрированное с ROS, для использования с DRONES (точнее, попугай бибоп). Сам код можно найти в Google, и в основном он показывает каждое лицо, которое появляется на экране. Моя проблема: я хочу, чтобы дрон следовал за моим лицом (и только моим), но, как я уже говорил, код может обнаруживать несколько лиц одновременно. Вот код:

face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
# some people can just use the haarcascade_frontalface_default.xml without specifying the path
test = face_cascade.load('../../../../opencv/data/haarcascades/haarcascade_frontalface_default.xml')

# start the video capture
video_capture = cv2.VideoCapture(0)

# while-loop to detect face on webcam until you press 'q'.
while not rospy.is_shutdown():
    # Capture frame-by-frame
    ret, frame = video_capture.read()
    frame = imutils.resize(frame, width=600)

    #convert the frame (of the webcam) to gray)
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    mask = cv2.erode(gray, None, iterations=2)
    mask = cv2.dilate(mask, None, iterations=2)

    # detecting the faces
    faces = face_cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=5, minSize=(30, 30), flags=cv2.CASCADE_SCALE_IMAGE)

    # Draw a rectangle around the faces
    for (x, y, w, h) in faces:
        cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 255, 0), 2)
        x_imagem = x + (w/2)
        y_imagem = y + (h/2)
        cv2.circle(frame, (x+(w/2), y+(h/2)), 5, (0,0,255), -1)
        #600 x 450;
        if(x_imagem > 200 and x_imagem < 400):
            rospy.loginfo("CENTRO")
        elif(x_imagem < 200): #ROSTO PRA ESQUERDA, ENTAO VAI PARA DIREITA
            rospy.loginfo("ROSTO NA ESQUERDA")
            pub_face.publish("esq")
        elif(x_imagem > 400): #ROSTO PRA DIREITA, ENTAO VAI PARA ESQUERDA
            rospy.loginfo("ROSTO NA DIREITA")
            pub_face.publish("dir")

    # Display the resulting frame
    cv2.imshow('Video', frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        rospy.loginfo("FIM DO PROGRAMA DE DETECCAO DE ROSTOS")
        break

# When everything is done, release the capture
video_capture.release()
cv2.destroyAllWindows()

Как видите, у нас есть массив FACES по адресу:

faces = face_cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=5, minSize=(30, 30), flags=cv2.CASCADE_SCALE_IMAGE)

Код нарисует прямоугольник там, где лицо, есть ли способ, которым я могу сделать так, чтобы он показывал только самый большой прямоугольник (в этом случае, мое лицо)? Надеюсь, я смог прояснить это!

Answer 1

Есть ли способ, чтобы я мог показать только самый большой прямоугольник (в этом случае, мое лицо)?

Выполните итерацию списка "граней" и вычислите площадь каждой области интереса (ROI), и захватите самую большую область.