main.py

from aip import AipFace
from picamera import PiCamera
import urllib.request
import RPi.GPIO as GPIO
import base64
import time
#Baidu API info Hiden
APP_ID = '22511984'
API_KEY = 'EqLIfjjW4I14AhDLTRebqVoF'
SECRET_KEY = 'HHIpusrNzWoFElGB0UZqQnpqQXBulgOM'
client = AipFace(APP_ID, API_KEY, SECRET_KEY)
#创建一个客户端用以访问百度云
#图像编码方式
IMAGE_TYPE='BASE64'
#camera = PiCamera()#定义一个摄像头对象
#用户组
GROUP = '01'


def setServoAngle(servo, angle):
    pwm = GPIO.PWM(servo, 50)
    pwm.start(8)
    dutyCycle = angle / 18. + 3.
    pwm.ChangeDutyCycle(dutyCycle)
    time.sleep(0.3)
    pwm.stop()


#照相函数
def getimage():
    camera.resolution = (1024,768)#摄像界面为1024*768
    camera.start_preview()#开始摄像
    time.sleep(2)
    camera.capture('faceimage.jpg')#拍照并保存
def opendoor():
    GPIO.setmode(GPIO.BCM)
    GPIO.setwarnings(False)
    pan = 22
    GPIO.setup(pan, GPIO.OUT) # gray ==> PAN
    pwm = GPIO.PWM(pan, 50)
    pwm.start(2.5)
    for i in range(0,70):
        dutyCycle = i / 18.0 + 2.5
        pwm.ChangeDutyCycle(dutyCycle)
        time.sleep(0.02)
    time.sleep(2)
    pwm.ChangeDutyCycle(0)
    GPIO.cleanup()

def closedoor():
    GPIO.setmode(GPIO.BCM)
    GPIO.setwarnings(False)
    pan = 22
    GPIO.setup(pan, GPIO.OUT) # gray ==> PAN
    pwm = GPIO.PWM(pan, 50)
    pwm.start(2.5)
    for i in range(50,-10):
        dutyCycle = i / 18.0 + 2.5
        pwm.ChangeDutyCycle(dutyCycle)
        time.sleep(0.02)
    time.sleep(2)
    pwm.ChangeDutyCycle(0)
    GPIO.cleanup()
#对图片的格式进行转换
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 == 'lhxone':
                print("欢迎%s !" % name)
                print(score)
            # if name == 'xiaoming':
            #     print("欢迎%s !" % name)
            #     time.sleep(3)
            # if name == "xiaoyu":
            #     print("欢迎%s !" % name)
        else:
            print("对不起，我不认识你！")
            name = 'Unknow'
            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
    if result['error_msg'] == 'pic not has face':
        print('检测不到人脸')
        time.sleep(2)
        return 0
    else:
        print(result['error_code']+' ' + result['error_code'])
        return 0
#主函数
if __name__ == '__main__':
    while True:
        print('准备')
        if True:
            getimage()#拍照
            img = transimage()#转换照片格式
            res = go_api(img)#将转换了格式的图片上传到百度云
            if(res == 1):#是人脸库中的人
                print("开")
                opendoor()
            else:
                print("关")
                closedoor()
            # time.slepp(3)
            #print('稍等三秒进入下一个')