通过百度云的AI开放平台的接口来实现的ai的人脸识别

image.png

安装环境：requirement.txt (pip install -r requirement.txt)

pymysql
sqlalchemy
tornado==4.5
baidu-aip

manage.py

import os
import tornado.web
import tornado.ioloop


from tornado.options import parse_command_line, options, define

from face.views import RegisterHandler, InitDbHandler, LogonHandler

define('port',default=80,type=int)



def make_app():
    return tornado.web.Application(handlers=[
        (r'/register/',RegisterHandler),
        (r'/init_db/', InitDbHandler),
        (r'/login/',LogonHandler)

    ],

    template_path=os.path.join(os.path.dirname(os.path.abspath(__file__)), 'templates'),
    static_path=os.path.join(os.path.dirname(os.path.abspath(__file__)), 'static'),
    )

if __name__ == '__main__':
    parse_command_line()
    app=make_app()
    app.listen(options.port)
    tornado.ioloop.IOLoop.current().start()

face/models.py

from datetime import datetime

from sqlalchemy import Column,Integer,String,DateTime

from untils.conn import Base

def init_db():
   Base.metadata.create_all()

class User(Base):
   id=Column(Integer,primary_key=True,autoincrement=True)
   username=Column(String(10), unique=True, nullable=False)
   realname = Column(String(10), unique=True, nullable=False)
   create_time = Column(DateTime, default=datetime.now)
   __tablename__ = 'user'

face/views.py

import tornado.web

from face.models import init_db, User
from untils.conn import Base, session
from untils.faceid import face_register, face_login


class RegisterHandler(tornado.web.RequestHandler):
    def get(self):
        error=''
        self.render('register.html', error=error)

    def post(self):
        #注册
        face=self.get_argument('face_img')

        img = face.split(',')[-1]
        username=self.get_argument('username')
        realname=self.get_argument('realname')
        if face and username and realname:
            #条用百度接口注册
            user=User()
            user.username = username
            user.realname = realname
            session.add(user)
            session.commit()

            res = face_register(img,user.id)
            if res:
                self.redirect('/login/')
            else:
                session.delete(user)
                session.commit()
                self.redirect('/register/')
        else:
            error = '请填写完整！'
            self.render('register.html', error=error)

class InitDbHandler(tornado.web.RequestHandler):
    def get(self):
        #将模型映射到数据库中
        #Base.metadata.create_all()
        init_db()
        self.write('创建表成功')
class LogonHandler(tornado.web.RequestHandler):
    def get(self):
        self.render('login.html')

    def post(self):
        face = self.get_argument('face_img')
        img = face.split(',')[-1]
        res = face_login(img)
        if res:
            self.write('成功')
        else:
            self.write('失败')

static/js/ai.js

var video = document.querySelector('video');
var audio, audioType;

var canvas1 = document.getElementById('canvas1');
var context1 = canvas1.getContext('2d');



navigator.getUserMedia = navigator.getUserMedia || navigator.webkitGetUserMedia || navigator.mozGetUserMedia || navigator.msGetUserMedia;
window.URL = window.URL || window.webkitURL || window.mozURL || window.msURL;

var exArray = []; //存储设备源ID
MediaStreamTrack.getSources(function (sourceInfos) {
    for (var i = 0; i != sourceInfos.length; ++i) {
        var sourceInfo = sourceInfos[i];
        //这里会遍历audio,video，所以要加以区分
        if (sourceInfo.kind === 'video') {
            exArray.push(sourceInfo.id);
        }
    }
});

function getMedia() {
    if (navigator.getUserMedia) {
        navigator.getUserMedia({
            'video': {
                'optional': [{
                    'sourceId': exArray[1] //0为前置摄像头，1为后置
                }]
            },
            'audio':true
        }, successFunc, errorFunc);    //success是获取成功的回调函数
    }
    else {
        alert('Native device media streaming (getUserMedia) not supported in this browser.');
    }
}

function successFunc(stream) {
    //alert('Succeed to get media!');
    if (video.mozSrcObject !== undefined) {
        //Firefox中，video.mozSrcObject最初为null，而不是未定义的，我们可以靠这个来检测Firefox的支持
        video.mozSrcObject = stream;
    }
    else {
        video.src = window.URL && window.URL.createObjectURL(stream) || stream;
    }

    //video.play();

    // 音频
    audio = new Audio();
    audioType = getAudioType(audio);
    if (audioType) {
        audio.src = 'polaroid.' + audioType;
        audio.play();
    }
}
function errorFunc(e) {
    alert('Error！'+e);
}

//获取音频格式
function getAudioType(element) {
    if (element.canPlayType) {
        if (element.canPlayType('audio/mp4; codecs="mp4a.40.5"') !== '') {
            return ('aac');
        } else if (element.canPlayType('audio/ogg; codecs="vorbis"') !== '') {
            return ("ogg");
        }
    }
    return false;
}

//拍照
function getPhoto() {
    context1.drawImage(video, 0, 0,220,220); //将video对象内指定的区域捕捉绘制到画布上指定的区域，实现拍照。
    var face = $('#face')
    var canvas1 = document.getElementById('canvas1');
    face.val(canvas1.toDataURL('image/png'))
}

templates/login.html(登陆界面)

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>登录</title>
    <script src="https://code.jquery.com/jquery-3.0.0.min.js"></script>

</head>
<body>
    <input type="button" value="摄像" onclick="getMedia()">
    <video src="" autoplay="autoplay" height="120px"></video>
    <canvas id="canvas1" height="220px" width="200"></canvas>
    <input type="button" value="拍照" height="120px" onclick="getPhoto();">
    <form action="" method="post">
        <input type="hidden" id="face" name="face_img">
        <p><input type="submit" value="提交"></p>

    </form>
    <script src="{{ static_url('js/ai.js') }}"></script>
</body>
</html>

templates/register.html(注册界面)

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>注册</title>
    <script src="https://code.jquery.com/jquery-3.0.0.min.js"></script>

</head>
<body>
    <input type="button" value="摄像" onclick="getMedia()">
    <video src="" autoplay="autoplay" height="120px"></video>
    <canvas id="canvas1" height="220px" width="200"></canvas>
    <input type="button" value="拍照" height="120px" onclick="getPhoto();">
    <form action="" method="post">
        <input type="hidden" id="face" name="face_img">
        <p>账号： <input type="text" name="username"></p>
        <p>真实姓名： <input type="text" name="realname"></p>
        <p><input type="submit" value="提交"></p>

    </form>
    <script src="{{ static_url('js/ai.js') }}"></script>
</body>
</html>

untils/conn.py

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

# 连接数据库格式
# mysql+pymysql://root:123456@127.0.0.1:3306/tornado9
db_url = 'mysql+pymysql://root:DENG5rong2hua0!@127.0.0.1:3306/tornado9'

# 创建引擎，建立连接
engine = create_engine(db_url)

# 模型与数据库表进行关联的基类，模型必须继承于Base
Base = declarative_base(bind=engine)

# 创建session会话
DbSession = sessionmaker(bind=engine)
session = DbSession()

untils/faceid.py

from aip import AipFace

""" 你的 APPID AK SK """
APP_ID = '15752252'
API_KEY = 'IwwxkcwUr6TBUYCfhCHD97e4'
SECRET_KEY = 'fKyhxKGR6P1XlP6iXfGzlVajagViHdgt'

client = AipFace(APP_ID, API_KEY, SECRET_KEY)

def face_register(image,userId,imageType='BASE64',groupId='user'):

    res = client.addUser(image, imageType,groupId,userId)
    if res['error_code']:
        return False
    return True
def face_login(image, imageType = 'BASE64', groupId = 'user'):
    res = client.search(image,imageType,groupId)
  #  print(res)
    if res['error_code']:
        return False
    return True