您好,登錄后才能下訂單哦!
這篇文章將為大家詳細講解有關基于face_recognition如何實現人臉識別,文章內容質量較高,因此小編分享給大家做個參考,希望大家閱讀完這篇文章后對相關知識有一定的了解。
準備工作
我們的人臉識別基于face_recognition庫。face_recognition基于dlib實現,用深度學習訓練數據,模型準確率高達99.38%。在開始我們的工作前,我們先安裝face_recognition
pip install face_recognition
人臉數字化
人臉識別的***步是檢測照片中的人臉區域,然后將人臉的圖像數據轉換成一個長度為128的向量,這128個數據代表了人臉的128個特征指標,如下所示
對于每一張已知人臉,生成這樣的一個128位的向量。對于一張未知人臉,將它的128位向量和所有已知人臉的128位向量一一比對,找到相似度***的那一個,即找出了未知人臉對應的人。
圖片數據
我們準備了兩張照片,分別是凱特王妃和威廉王子的單人照,分別存成catherine.jpg和william.jpg,這兩張照片中的人臉作為我們的已知人臉
我們的目標是在下面的合影中識別出兩人的臉并在圖中標出各自的名字。下圖存成unknown.jpg
代碼實現
接下來開始我們的編程工作
import cv2 import face_recognitionnames = [ "catherine", "william", ]
首先我們定義了標簽集,存在names數組中。
標簽名字也是我們圖片的文件名。
images = [] for name in names: filename = name + ".jpg" image = face_recognition.load_image_file(filename) images.append(image) unknown_image = face_recognition.load_image_file("unknown.jpg")
調用face_recognition.load_image_file從圖片中讀取數據。
這里讀取了包含已知人臉和未知人臉的圖片的數據,未知人臉的圖片就是上面的合影圖片unknown.jpg。
face_encodings = [] for image in images: encoding = face_recognition.face_encodings(image)[0] face_encodings.append(encoding) unknown_face_encodings = face_recognition.face_encodings(unknown_image)
face_recognition.face_encodings會返回圖片中的所有的人臉的128位向量。單人照片只有一張人臉,所以face_recognition.face_encodings(image)[0]只取***個元素。合影圖片中包含了2張人臉,所以unknown_face_encodings包含2個128位向量。
face_locations = face_recognition.face_locations(unknown_image) for i in range(len(unknown_face_encodings)): unknown_encoding = unknown_face_encodings[i] face_location = face_locations[i] top, right, bottom, left = face_location cv2.rectangle(unknown_image, (left, top), (right, bottom), (0, 255, 0), 2) results = face_recognition.compare_faces(face_encodings, unknown_encoding) for j in range(len(results)): if results[j]: name = names[j] cv2.putText(unknown_image, name, (left-10, top-10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2) unknown_image_rgb = cv2.cvtColor(unknown_image, cv2.COLOR_BGR2RGB) cv2.imshow("Output", unknown_image_rgb) cv2.waitKey(0)
face_locations存了每張臉的位置信息。
在循環中我們調用cv2.rectangle框出了檢測到的每張臉。
face_recognition.compare_faces將已知人臉的128位向量和每張未知人臉的128位向量做比較,結果存入results數組中。results數組中的每一個元素都是True或者False,長度和人臉個數相等。results中的每個元素都和已知人臉一一對應,在某一個位置處的元素為True,表示未知人臉被識別成這張已知人臉。
對識別出來的每張人臉,我們調用cv2.putText在圖上標注標簽。
以上是代碼的全部。
測試
令人興奮的時刻又來到了! 我們來檢驗一下我們的成果。
運行上面的程序,可以看到下面的結果
威廉王子和凱特王妃的人臉都被準確地識別出來。綠色的框框出了人臉區域,框的上方標注了識別到的人的名字。
關于基于face_recognition如何實現人臉識別就分享到這里了,希望以上內容可以對大家有一定的幫助,可以學到更多知識。如果覺得文章不錯,可以把它分享出去讓更多的人看到。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。