from aip import AipFace
from picamera import PiCamera
import urllib.request
import RPi.GPIO as GPIO
import base64
import time
#百度人脸识别API账号信息
APP_ID = '换成自己的APP_ID'
API_KEY = '换成自己的API_KEY'
SECRET_KEY ='换成自己的SECRET_KEY'
client = AipFace(APP_ID, API_KEY, SECRET_KEY)#创建一个客户端用以访问百度云
#图像编码方式
IMAGE_TYPE='BASE64'
camera = PiCamera()#定义一个摄像头对象
#用户组
GROUP = '换成自己的组别 '
#照相函数
def getimage():
camera.resolution = (1024,768)#摄像界面为1024*768
camera.start_preview()#开始摄像
camera.capture('faceimage.jpg')#拍照并保存
#对图片的格式进行转换
def transimage():
f = open('faceimage.jpg','rb')
img = base64.b64encode(f.read())
return img
#上传到百度api进行人脸检测
def go_api(image):
result = client.search(str(image, 'utf-8'), IMAGE_TYPE, GROUP);#在百度云人脸库中寻找有没有匹配的人脸
if result['error_msg'] == 'SUCCESS':#如果成功了
name = result['result']['user_list'][0]['user_id']#获取名字
score = result['result']['user_list'][0]['score']#获取相似度
if score > 80:#如果相似度大于80
if name == 'GeYuYao':
print("Welcome %s !" % name)
time.sleep(1)
return 0
curren_time = time.asctime(time.localtime(time.time()))#获取当前时间
#将人员出入的记录保存到Log.txt中
f = open('Log.txt','a')
f.write("Person: " + name + " " + "Time:" + str(curren_time)+'\n')
f.close()
return 1
else:
print('test falut')
time.sleep(1)
return 0
#主函数
if __name__ == '__main__':
while True:
print('ready')
time.sleep(1)
getimage() # 拍照
img = transimage() # 转换照片格式
res = go_api(img) # 将转换了格式的图片上传到百度云
print('The round is over')
time.sleep(1)
树莓派人脸识别源代码
最新推荐文章于 2024-05-15 02:23:17 发布