找失踪孩子——人脸识别

我在用人脸识别找一个小孩
https://bbs.baobeihuijia.com/thread-218332-1-1.html


搜索用的图片都是网上别人的旅游、逛街拍的图片等等，但图片还是不够多，我在想一个大家一起找的方法：
一个志愿者上班、旅游、逛街等等时，多拍一些周边的照片，回到家对照片进行识别，
照片分享给其他人进行识别。


程序流程

1. import face_recognition
   
2. import cv2
   
3. import os
   
4. import shutil
   
5. 
   
6. """
   
7. 根据失踪儿童照片所在目录known_children，获取人脸特征
   
8. 匹配判断unknown_pic目录中人脸
   
9.     a1.jpg a2.jpg a3.jpg 是匹配的同一个人
   
10.     图片来源可以是摄像头拍的照片、百度微博上爬取（比如搜索 街道 热闹）
    
11. 结果输出到result目录（匹配的图片以 'A_' 前缀）
    
12. 处理后的图片放入finish目录
    
13. """
    
14. 
    
15. # 当前目录
    
16. GlobalDir = os.getcwd()
    
17. # 储存失踪儿童的名字，取自文件名（不要使用中文，会乱码。需要的自行修改代码），用于标注
    
18. names = []
    
19. # 失踪儿童的脸型数据
    
20. face_encodings = []
    
21. # 脸型数据匹配的阈值
    
22. recognizeTolerance=0.50
    
23. 
    
24. 
    
25. def recognizeFace(unknown_image,filename, outputUnknow=False):
    
26.     """
    
27.     :param unknown_image: 需要进行判断的图片
    
28.     :param outputUnknow: 未找到匹配时，是否还输出图片
    
29.     """
    
30. 
    
31.     print(filename)
    
32. 
    
33.     unknown_face_encodings = face_recognition.face_encodings(unknown_image)
    
34. 
    
35.     face_locations = face_recognition.face_locations(unknown_image)
    
36. 
    
37.     finded=False
    
38.     for i in range(len(unknown_face_encodings)):
    
39. 
    
40.         unknown_encoding = unknown_face_encodings[i]
    
41. 
    
42.         face_location = face_locations[i]
    
43. 
    
44.         top, right, bottom, left = face_location
    
45. 
    
46.         results = face_recognition.compare_faces(face_encodings, unknown_encoding, tolerance=recognizeTolerance)
    
47.         for j in range(len(results)):
    
48.             if results[j]:
    
49.                 finded=True
    
50.                 name = names[j]
    
51.                 cv2.putText(unknown_image, name, (left - 10, top - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
    
52. 
    
53.         cv2.rectangle(unknown_image, (left, top), (right, bottom), (0, 255, 0), 2)
    
54. 
    
55.     if (outputUnknow or finded):
    
56. 
    
57.         unknown_image_rgb = cv2.cvtColor(unknown_image, cv2.COLOR_BGR2RGB)
    
58. 
    
59.         # filename = time.strftime('%m%d_%H%M%S', time.localtime(time.time()))
    
60.         if finded:
    
61.             filename = 'A_'+filename # 有匹配脸型的文件名前缀 A
    
62. 
    
63.         cv2.imwrite(GlobalDir + "\\result\" + filename, unknown_image_rgb)
    
64. 
    
65. 
    
66. def main():
    
67. 
    
68.     rootdir = GlobalDir + '\\known_children'
    
69.     list = os.listdir(rootdir)
    
70.     for i in range(0, len(list)):
    
71.         path = os.path.join(rootdir, list[i])
    
72.         if os.path.isfile(path) and path.endswith('.jpg'):
    
73.             image = face_recognition.load_image_file(path)
    
74.             # images.append(image)
    
75.             encoding = face_recognition.face_encodings(image)[0]
    
76.             face_encodings.append(encoding)
    
77. 
    
78.             names.append(path.split("\")[-1].split(".")[0])
    
79. 
    
80.     # unknown_image = face_recognition.load_image_file(GlobalDir + "\\unknown_pic\\a4.jpg")
    
81.     # analyzeFace(unknown_image, False)
    
82. 
    
83.     rootdir = GlobalDir+'\\unknown_pic'
    
84.     list = os.listdir(rootdir)  # 列出文件夹下所有的目录与文件
    
85.     for i in range(0, len(list)):
    
86.         path = os.path.join(rootdir, list[i])
    
87.         if os.path.isfile(path) and path.endswith('.jpg'):
    
88.             unknown_image = face_recognition.load_image_file(path)
    
89.             recognizeFace(unknown_image,path.split("\")[-1], False)
    
90.             shutil.move(path, GlobalDir + '\\finish\\'+path.split("\")[-1])
    
91. 
    
92. if __name__ == '__main__':
    
93.     main()

复制代码