資源簡介
Python直接使用模型,實現一個簡易版的人臉對比,非常有趣!
代碼片段和文件信息
#?-*-?coding:?utf-8?-*-
import?osdlibglobnumpy
from?skimage?import?io
#?人臉關鍵點檢測器
predictor_path?=?“shape_predictor.dat“
#?人臉識別模型、提取特征值
face_rec_model_path?=?“dlib_face_recognition.dat“
#?訓練圖像文件夾
faces_folder_path?=‘train_images‘?
#?加載模型
detector?=?dlib.get_frontal_face_detector()
sp?=?dlib.shape_predictor(predictor_path)
facerec?=?dlib.face_recognition_model_v1(face_rec_model_path)
candidate?=?[]?????????#?存放訓練集人物名字
descriptors?=?[]???????#存放訓練集人物特征列表
for?f?in?glob.glob(os.path.join(faces_folder_path“*.jpg“)):
????print(“正在處理:?{}“.format(f))
????img?=?io.imread(f)
????candidate.append(f.split(‘\\‘)[-1].split(‘.‘)[0])
????#?人臉檢測
????dets?=?detector(img?1)
????for?k?d?in?enumerate(dets):?
????????shape?=?sp(img?d)
????????#?提取特征
????????face_descriptor?=?facerec.compute_face_descriptor(img?shape)
????????v?=?numpy.array(face_descriptor)?
????????descriptors.append(v)
print(‘識別訓練完畢!‘)
try:
##????test_path=input(‘請輸入要檢測的圖片的路徑(記得加后綴哦):‘)
????img?=?io.imread(r“.\test_images\test6.jpg“)
????dets?=?detector(img?1)
except:
????print(‘輸入路徑有誤,請檢查!‘)
dist?=?[]
for?k?d?in?enumerate(dets):
????shape?=?sp(img?d)
????face_descriptor?=?facerec.compute_face_descriptor(img?shape)
????d_test?=?numpy.array(face_descriptor)?
????for?i?in?descriptors:????????????????#計算距離
????????dist_?=?numpy.linalg.norm(i-d_test)
????????dist.append(dist_)
#?訓練集人物和距離組成一個字典
c_d?=?dict(zip(candidatedist))????????????????
cd_sorted?=?sorted(c_d.items()?key=lambda?d:d[1])
print?(“識別到的人物最有可能是:?“cd_sorted[0][0])
print(c_d)
?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????文件???????6148??2018-11-06?19:49??簡易人臉相似度對比\face_recognition1106\.DS_Store
?????文件???22466066??2017-10-05?19:15??簡易人臉相似度對比\face_recognition1106\dlib_face_recognition.dat
?????文件???????1843??2018-11-08?15:57??簡易人臉相似度對比\face_recognition1106\Face_recognition.py
?????文件???99693937??2017-10-04?20:18??簡易人臉相似度對比\face_recognition1106\shape_predictor.dat
?????文件??????37271??2017-10-05?20:45??簡易人臉相似度對比\face_recognition1106\test_images\test1.jpg
?????文件?????146414??2018-10-27?21:22??簡易人臉相似度對比\face_recognition1106\test_images\test10.jpg
?????文件??????43036??2017-10-08?19:13??簡易人臉相似度對比\face_recognition1106\test_images\test2.jpg
?????文件??????39212??2017-10-08?18:48??簡易人臉相似度對比\face_recognition1106\test_images\test3.jpg
?????文件??????44667??2017-10-08?18:59??簡易人臉相似度對比\face_recognition1106\test_images\test4.jpg
?????文件??????11836??2017-10-09?20:52??簡易人臉相似度對比\face_recognition1106\test_images\test5.jpg
?????文件??????42265??2017-10-11?16:20??簡易人臉相似度對比\face_recognition1106\test_images\test6.jpg
?????文件??????10283??2017-10-09?21:15??簡易人臉相似度對比\face_recognition1106\test_images\test7.jpg
?????文件??????41181??2017-10-11?15:43??簡易人臉相似度對比\face_recognition1106\test_images\test8.jpg
?????文件??????44170??2017-10-11?15:48??簡易人臉相似度對比\face_recognition1106\test_images\test9.jpg
?????文件??????18579??2017-10-05?19:30??簡易人臉相似度對比\face_recognition1106\train_images\劉亦菲.jpg
?????文件??????30191??2017-10-05?19:30??簡易人臉相似度對比\face_recognition1106\train_images\劉詩詩.jpg
?????文件??????14306??2018-10-27?21:20??簡易人臉相似度對比\face_recognition1106\train_images\唐藝昕.jpg
?????文件??????19743??2017-10-09?20:58??簡易人臉相似度對比\face_recognition1106\train_images\楊冪.jpg
?????文件??????34356??2017-10-09?16:51??簡易人臉相似度對比\face_recognition1106\train_images\迪麗熱巴.jpg
?????文件???????7298??2017-10-11?15:41??簡易人臉相似度對比\face_recognition1106\train_images\鄭秀妍.jpg
?????文件??????23112??2017-10-11?15:40??簡易人臉相似度對比\face_recognition1106\train_images\鄭秀晶.jpg
?????文件??????16772??2017-10-05?19:36??簡易人臉相似度對比\face_recognition1106\train_images\郭富城.jpg
?????文件??????23841??2017-10-08?18:57??簡易人臉相似度對比\face_recognition1106\train_images\黎明.jpg
?????目錄??????????0??2018-11-03?15:58??簡易人臉相似度對比\face_recognition1106\test_images
?????目錄??????????0??2018-11-03?16:08??簡易人臉相似度對比\face_recognition1106\train_images
?????目錄??????????0??2018-11-06?19:49??簡易人臉相似度對比\face_recognition1106
?????目錄??????????0??2018-11-08?16:00??簡易人臉相似度對比
-----------?---------??----------?-----??----
????????????122816527????????????????????27
............此處省略0個文件信息
評論
共有 條評論