Advertisement
franciscominajas

ReconocimientoRostros

Oct 26th, 2019
183
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 2.66 KB | None | 0 0
  1. import face_recognition
  2. import cv2
  3. import numpy as np
  4.  
  5. # Obtenemos la camara
  6. video_capture = cv2.VideoCapture(0)
  7.  
  8. # Cargamos una imagen y aprendemos a reconocerla
  9. obama_image = face_recognition.load_image_file("obama.jpg")
  10. obama_face_encoding = face_recognition.face_encodings(obama_image)[0]
  11.  
  12. # Cargamos una segunda imagen y aprendemos a reconocer
  13. biden_image = face_recognition.load_image_file("biden.jpg")
  14. biden_face_encoding = face_recognition.face_encodings(biden_image)[0]
  15.  
  16. # Creamos los nombres de las caras a reconcer
  17. known_face_encodings = [
  18.     obama_face_encoding,
  19.     biden_face_encoding
  20. ]
  21. known_face_names = [
  22.     "Barack Obama",
  23.     "Joe Biden"
  24. ]
  25.  
  26. while True:
  27.     # Grabamos un video
  28.     ret, frame = video_capture.read()
  29.  
  30.     # Convert the image from BGR color (which OpenCV uses) to RGB color (which face_recognition uses)
  31.     rgb_frame = frame[:, :, ::-1]
  32.  
  33.     # Convierta la imagen de color BGR (que utiliza OpenCV) a color RGB (que utiliza face_recognition)
  34.     face_locations = face_recognition.face_locations(rgb_frame)
  35.     face_encodings = face_recognition.face_encodings(rgb_frame, face_locations)
  36.  
  37.     # Recorre cada cara en este cuadro de video
  38.     for (top, right, bottom, left), face_encoding in zip(face_locations, face_encodings):
  39.         # Vea si la cara coincide con las caras conocidas
  40.         matches = face_recognition.compare_faces(known_face_encodings, face_encoding)
  41.  
  42.         name = "Unknown"
  43.  
  44.         # Si se encontró una coincidencia en face_encodings conocidos, solo use la primera.
  45.         # si es cierto en los partidos:
  46.         #     first_match_index = match.index (True)
  47.         #     name = known_face_names[first_match_index]
  48.  
  49.         # O en su lugar, use la cara conocida con la menor distancia a la nueva cara
  50.         face_distances = face_recognition.face_distance(known_face_encodings, face_encoding)
  51.         best_match_index = np.argmin(face_distances)
  52.         if matches[best_match_index]:
  53.             name = known_face_names[best_match_index]
  54.  
  55.         # Dibuja un cuadro alrededor de la cara
  56.         cv2.rectangle(frame, (left, top), (right, bottom), (0, 0, 255), 2)
  57.  
  58.         # Dibuja una etiqueta con un nombre debajo de la cara
  59.         cv2.rectangle(frame, (left, bottom - 35), (right, bottom), (0, 0, 255), cv2.FILLED)
  60.         font = cv2.FONT_HERSHEY_DUPLEX
  61.         cv2.putText(frame, name, (left + 6, bottom - 6), font, 1.0, (255, 255, 255), 1)
  62.  
  63.     # Mostrar la imagen resultante
  64.     cv2.imshow('Video', frame)
  65.  
  66.     # ¡Presiona 'q' en el teclado para salir!
  67.     if cv2.waitKey(1) & 0xFF == ord('q'):
  68.         break
  69.  
  70. # Suelte el asa a la cámara web
  71. video_capture.release()
  72. cv2.destroyAllWindows()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement