Initial commit

OpenCV/tutoriel8-2/identifie.py

@@ -0,0 +1,48 @@
+#!/usr/bin/env python
+import cv2
+import pickle
+import numpy as np
+import common as c
+
+face_cascade= cv2.CascadeClassifier('haarcascade_frontalface_alt2.xml')
+recognizer=cv2.face.LBPHFaceRecognizer_create()
+recognizer.read("trainner.yml")
+id_image=0
+color_info=(255, 255, 255)
+color_ko=(0, 0, 255)
+color_ok=(0, 255, 0)
+
+with open("labels.pickle", "rb") as f:
+    og_labels=pickle.load(f)
+    labels={v:k for k, v in og_labels.items()}
+
+cap=cv2.VideoCapture("Plan 9 from Outer Space Charles Burg, J. Edward Reynolds, Hu.mp4")
+while True:
+    ret, frame=cap.read()
+    tickmark=cv2.getTickCount()
+    gray=cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
+    faces=face_cascade.detectMultiScale(gray, scaleFactor=1.2,minNeighbors=4, minSize=(c.min_size, c.min_size))
+    for (x, y, w, h) in faces:
+        roi_gray=cv2.resize(gray[y:y+h, x:x+w], (c.min_size, c.min_size))
+        id_, conf=recognizer.predict(roi_gray)
+        if conf<=95:
+             color=color_ok
+             name=labels[id_]
+        else:
+            color=color_ko
+            name="Inconnu"
+        label=name+" "+'{:5.2f}'.format(conf)
+        cv2.putText(frame, label, (x, y-10), cv2.FONT_HERSHEY_DUPLEX, 1, color_info, 1, cv2.LINE_AA)
+        cv2.rectangle(frame, (x, y), (x+w, y+h), color, 2)
+    fps=cv2.getTickFrequency()/(cv2.getTickCount()-tickmark)
+    cv2.putText(frame, "FPS: {:05.2f}".format(fps), (10, 30), cv2.FONT_HERSHEY_PLAIN, 2, color_info, 2)
+    cv2.imshow('L42Project', frame)
+    key=cv2.waitKey(1)&0xFF
+    if key==ord('q'):
+        break
+    if key==ord('a'):
+        for cpt in range(100):
+            ret, frame=cap.read()
+
+cv2.destroyAllWindows()
+print("Fin")