import?numpy?as?np
import?cv2
import?time
import?face_recognition

#?Threshold?=?0.65?#?人臉置信度閾值

‘‘‘
功能：計算兩張圖片的相似度，范圍：[01]
輸入：
	1）人臉A的特征向量
	2）人臉B的特征向量
輸出：
	1）sim：AB的相似度
‘‘‘
def?simcos（AB）:
	A=np.array（A）
	B=np.array（B）
	dist?=?np.linalg.norm（A?-?B）?#?二范數
	sim?=?1.0?/?（1.0?+?dist）?#
	return?sim


‘‘‘
功能：
輸入：
	1）x:人臉庫向量（n維）
	2）y：被測人臉的特征向量（1維）
輸出：
	1）match：與人臉庫匹配列表，如[FalseTrueTrueFalse]
			?表示被測人臉y與人臉庫x的第23張圖片匹配，與14不匹配
	2）max（ressim）:最大相似度
‘‘‘
def?compare_faces（xyThreshold）:
	ressim?=?[]
	match?=?[False]*len（x）
	for?fet?in?x:
		sim?=?simcos（fety）
		ressim.append（sim）
	if?max（ressim）?>?Threshold:??#置信度閾值
		match[ressim.index（max（ressim））]?=?True
	return?matchmax（ressim）


‘‘‘
注冊身份
輸入：
	1）libpath：人臉庫地址
輸出：
	1）known_face_encodings：人臉庫特征向量
	2）known_face_names：人臉庫名字標簽
‘‘‘
def?registeredIdentity（libpath）:
	known_face_encodings?known_face_names?=?[]?[]
	with?open（libpath?+?‘liblist.txt‘?‘r‘）?as?f:
		lines?=?f.readlines（）
	for?line?in?lines:
		img_lable_name?=?line.split（）
		image?=?face_recognition.load_image_file（libpath?+?str（img_lable_name[0]））
		face_locations?=?face_recognition.face_locations（image）
		#?face_locations?=?face_recognition.face_locations（image?model=‘cnn‘）

		face_encoding?=?face_recognition.face_encodings（image?face_locations）[0]
		#?face_encoding?=?face_recognition.face_encodings（image?face_locations）
		known_face_encodings.append（face_encoding）
		known_face_names.append（str（img_lable_name[1]））
	return?known_face_encodings?known_face_names


‘‘‘
輸入：
	1）testimg：測試圖片
	2）known_face_encodings：人臉庫特征向量
	3）known_face_names：人臉庫名字標簽
輸出：
	1）retname：預測的名字
	2）retscore：相似度得分
	3）face_locations：人臉位置坐標
‘‘‘
def?identityRecognition（testimgknown_face_encodingsknown_face_namesThreshold）:
	face_locations?=?face_recognition.face_locations（testimg）
	#?face_locations?=?face_recognition.face_locations（testimg?model=“cnn“）
	face_encodings?=?face_recognition.face_encodings（testimg?face_locations）
	retname?retscore?=?“Noface“?0
	for?face_encoding?in?face_encodings:
		matches?score?=?compare_faces（known_face_encodings?face_encodingThreshold）
		retname?retscore?=?“Unknow“?0
		if?True?in?matches:
			first_match_index?=?matches.index（True）
			name?=?known_face_names[first_match_index]
			if?score?>?retscore:
				retname?=?name
				retscore?=?score
	return?retname?retscoreface_locations


‘‘‘
輸入：
	1）img:攝像頭得到的未裁剪圖片
	2）face_locations:人臉位置坐標
	3）?name:預測的名字
輸出：
	img:加框加年齡備注之后的畫面
‘‘‘
def?age_show（img??face_locationsname）:
	for?（y0?x1?y1?x0）?in?face_locations:
		cv2.rectangle（img?（x0?y0）?（x1?y1）?（?0?0255）?2）
		info?=?str（name）
		t_size?=?cv2.getTextSize（str（info）?cv2.FONT_HERSHEY_PLAIN?1?2）[0]
		x2y2?=?x0?+?t_size[0]?+?3?y0?+?t_size[1]?+?4
		cv2.rectangle（img?（x0y0）?（x2y2）?（0?0?255）?-1）??#?-1填充作為文字框底色
		cv2.putText（img?info?（x0?y0?+t_size[1]+4）?cv2.FONT_HERSHEY_PLAIN?1?（0?0?0）?1）
	return?img

?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2018-08-07?15:15??.idea\
?????文件?????????455??2018-08-07?13:10??.idea\face_recognition.iml
?????目錄???????????0??2018-08-07?13:10??.idea\inspectionProfiles\
?????文件?????????288??2018-08-07?13:10??.idea\misc.xml
?????文件?????????284??2018-08-07?13:10??.idea\modules.xml
?????文件???????16408??2018-08-07?15:15??.idea\workspace.xml
?????目錄???????????0??2018-08-07?14:18??__pycache__\
?????文件????????2095??2018-08-07?14:18??__pycache__\face.cpython-36.pyc
?????文件????????3526??2018-08-07?14:27??face.py
?????目錄???????????0??2018-08-07?14:52??facelib\
?????文件??????145405??2018-06-28?17:54??facelib\1.png
?????文件???????31910??2018-06-28?20:31??facelib\11.jpg
?????文件??????865334??2018-06-28?17:54??facelib\11.png
?????文件???????18128??2018-08-07?14:52??facelib\55.jpg
?????文件??????????58??2018-08-07?14:52??facelib\liblist.txt
?????文件?????????881??2018-08-07?14:27??pic_demo.py
?????目錄???????????0??2018-08-07?13:15??Test\
?????目錄???????????0??2018-08-07?13:16??Test\pic_in\
?????文件???????43544??2018-06-26?10:52??Test\pic_in\test1.png
?????目錄???????????0??2018-08-07?13:27??Test\pic_out\
?????文件??????203243??2018-08-07?14:12??Test\pic_out\test1.png
?????目錄???????????0??2018-08-07?13:15??Test\video_in\
?????目錄???????????0??2018-08-07?14:19??Test\video_out\
?????文件??????202210??2018-08-07?14:26??Test\video_out\1.avi
?????文件????????1467??2018-08-07?14:26??video_demo.py