Initial commit

OpenCV/tutoriel8-2/README.md

@@ -0,0 +1,6 @@
+# Tutoriel OpenCV
+## Identification avec face.LBPHFaceRecognizer partie 2
+
+La vidéo du tutoriel est à l'adresse:
+https://www.youtube.com/watch?v=UNZ06RZRTUQ
+

OpenCV/tutoriel8-2/apprentissage.py

@@ -0,0 +1,38 @@
+import cv2
+import os
+import numpy as np
+import pickle
+import common as c
+
+image_dir="./photos/"
+current_id=0
+label_ids={}
+x_train=[]
+y_labels=[]
+
+for root, dirs, files in os.walk(image_dir):
+    if len(files):
+        label=root.split("/")[-1]
+        for file in files:
+            if file.endswith("png"):
+                path=os.path.join(root, file)
+                if not label in label_ids:
+                    label_ids[label]=current_id
+                    current_id+=1
+                id_=label_ids[label]
+                image=cv2.resize(cv2.imread(path, cv2.IMREAD_GRAYSCALE), (c.min_size, c.min_size))
+                fm=cv2.Laplacian(image, cv2.CV_64F).var()
+                if fm<250:
+                    print("Photo exclue:", path, fm)
+                else:
+                    x_train.append(image)
+                    y_labels.append(id_)
+
+with open("labels.pickle", "wb") as f:
+    pickle.dump(label_ids, f)
+
+x_train=np.array(x_train)
+y_labels=np.array(y_labels)
+recognizer=cv2.face.LBPHFaceRecognizer_create()
+recognizer.train(x_train, y_labels)
+recognizer.save("trainner.yml")

OpenCV/tutoriel8-2/common.py

@@ -0,0 +1 @@
+min_size=70

OpenCV/tutoriel8-2/enregistrement_visage.py

@@ -0,0 +1,28 @@
+import cv2
+import operator
+import common as c
+
+face_cascade=cv2.CascadeClassifier("./haarcascade_frontalface_alt2.xml")
+#cap=cv2.VideoCapture("Plan 9 from Outer Space Charles Burg, J. Edward Reynolds, Hu.mp4")
+cap=cv2.VideoCapture("Plan_9_from_Outer_Space_1959_512kb.mp4")
+
+id=0
+while True:
+    ret, frame=cap.read()
+    gray=cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
+    face=face_cascade.detectMultiScale(gray, scaleFactor=1.2, minNeighbors=4, minSize=(c.min_size, c.min_size))
+    for x, y, w, h in face:
+        cv2.imwrite("non-classees/p-{:d}.png".format(id), frame[y:y+h, x:x+w])
+        cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 0, 255), 2)
+        id+=1
+    cv2.imshow('video', frame)
+    key=cv2.waitKey(1)&0xFF
+    if key==ord('q'):
+        break
+    if key==ord('a'):
+        for cpt in range(100):
+            ret, frame=cap.read()
+    for cpt in range(4):
+        ret, frame=cap.read()
+cap.release()
+cv2.destroyAllWindows()

OpenCV/tutoriel8-2/enregistrement_visage_V2.py

@@ -0,0 +1,46 @@
+import cv2
+import operator
+import os
+import common as c
+
+video="Plan_9_from_Outer_Space_1959_512kb.mp4"
+cascade="./haarcascade_frontalface_alt2.xml"
+img_non_classees='non-classees'
+
+if not os.path.exists(video):
+    print("Le fichier video n'existe pas", video)
+    quit()
+
+if not os.path.exists(cascade):
+    print("Le fichier cascade n'existe pas", cascade)
+    quit()
+
+face_cascade=cv2.CascadeClassifier(cascade)
+cap=cv2.VideoCapture(video)
+
+if not os.path.isdir(img_non_classees):
+    os.mkdir(img_non_classees)
+    
+id=0
+while True:
+    ret, frame=cap.read()
+    if ret is False:
+        break
+    gray=cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
+    face=face_cascade.detectMultiScale(gray, scaleFactor=1.2, minNeighbors=4, minSize=(c.min_size, c.min_size))
+    for x, y, w, h in face:
+        cv2.imwrite("{}/p-{:d}.png".format(img_non_classees, id), frame[y:y+h, x:x+w])
+        cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 0, 255), 2)
+        id+=1
+    cv2.imshow('video', frame)
+    key=cv2.waitKey(1)&0xFF
+    if key==ord('q'):
+        break
+    if key==ord('a'):
+        for cpt in range(100):
+            ret, frame=cap.read()
+    for cpt in range(4):
+        ret, frame=cap.read()
+
+cap.release()
+cv2.destroyAllWindows()

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")