SpringBoot集成百度人脸识别实现登陆注册功能Demo

一、前言

刚进入新公司没几天，领导让我写一个人脸识别的demo用来集成到现有的软件中，今天这篇博客用来记录一下这个demo的实现过程

二、项目准备

我们需要注册一个百度人脸识别云服务的账号并完成个人认证领取免费资源【百度人脸识别云服务官网】

注册完成后领取免费资源（前提是要完成个人认证）

领取完资源后我们创建一个新应用，根据自己的项目需求勾选服务

创建完成后我们会得到API Key和Secret Key

三、代码编写

代码很简单，我们照抄官方API文档
项目结构：

3.1 导入SDK

<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd"><modelVersion>4.0.0modelVersion><parent><groupId>org.springframework.bootgroupId><artifactId>spring-boot-starter-parentartifactId><version>2.6.5version><relativePath/> parent><groupId>com.jzjgroupId><artifactId>FaceRecognitionDemoartifactId><version>0.0.1-SNAPSHOTversion><properties><java.version>1.8java.version>properties><dependencies><dependency><groupId>com.baidu.aipgroupId><artifactId>java-sdkartifactId><version>4.9.0version>dependency><dependency><groupId>com.google.code.gsongroupId><artifactId>gsonartifactId><version>2.8.5version>dependency><dependency><groupId>com.alibabagroupId><artifactId>fastjsonartifactId><version>1.2.79version>dependency><dependency><groupId>org.springframework.bootgroupId><artifactId>spring-boot-starter-webartifactId>dependency><dependency><groupId>org.springframework.bootgroupId><artifactId>spring-boot-starter-testartifactId><scope>testscope>dependency><dependency><groupId>org.projectlombokgroupId><artifactId>lombokartifactId>dependency>dependencies><build><plugins><plugin><groupId>org.springframework.bootgroupId><artifactId>spring-boot-maven-pluginartifactId>plugin>plugins>build>project>

3.2 获取AccessToken

• 向API服务地址使用POST发送请求，必须在URL中带上参数access_token，可通过后台的API Key和Secret Key生成

package com.baidu.ai.aip.auth;import org.json.JSONObject;import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.List;
import java.util.Map;/*** 获取token类*/
public class AuthService {/*** 获取权限token* @return 返回示例：* {* "access_token": "24.460da4889caad24cccdb1fea17221975.2592000.1491995545.282335-1234567",* "expires_in": 2592000* }*/public static String getAuth() {// 官网获取的 API Key 更新为你注册的String clientId = "百度云应用的AK";// 官网获取的 Secret Key 更新为你注册的String clientSecret = "百度云应用的SK";return getAuth(clientId, clientSecret);}/*** 获取API访问token* 该token有一定的有效期，需要自行管理，当失效时需重新获取.* @param ak - 百度云官网获取的 API Key* @param sk - 百度云官网获取的 Securet Key* @return assess_token 示例：* "24.460da4889caad24cccdb1fea17221975.2592000.1491995545.282335-1234567"*/public static String getAuth(String ak, String sk) {// 获取token地址String authHost = "https://aip.baidubce.com/oauth/2.0/token?";String getAccessTokenUrl = authHost// 1. grant_type为固定参数+ "grant_type=client_credentials"// 2. 官网获取的 API Key+ "&client_id=" + ak// 3. 官网获取的 Secret Key+ "&client_secret=" + sk;try {URL realUrl = new URL(getAccessTokenUrl);// 打开和URL之间的连接HttpURLConnection connection = (HttpURLConnection) realUrl.openConnection();connection.setRequestMethod("GET");connection.connect();// 获取所有响应头字段Map<String, List<String>> map = connection.getHeaderFields();// 遍历所有的响应头字段for (String key : map.keySet()) {System.err.println(key + "--->" + map.get(key));}// 定义 BufferedReader输入流来读取URL的响应BufferedReader in = new BufferedReader(new InputStreamReader(connection.getInputStream()));String result = "";String line;while ((line = in.readLine()) != null) {result += line;}/*** 返回结果示例*/System.err.println("result:" + result);JSONObject jsonObject = new JSONObject(result);String access_token = jsonObject.getString("access_token");return access_token;} catch (Exception e) {System.err.printf("获取token失败！");e.printStackTrace(System.err);}return null;}
}

注意：access_token的有效期为30天，切记需要每30天进行定期更换，或者每次请求都拉取新token

3.3 人脸注册

向人脸库中添加人脸

• 注意事项：
  • 请求体格式化：Content-Type为application/json，通过json格式化请求体。
  • Base64编码：请求的图片需经过Base64编码，图片的base64编码指将图片数据编码成一串字符串，使用该字符串代替图像地址。您可以首先得到图片的二进制，然后用Base64格式编码即可。需要注意的是，图片的base64编码是不包含图片头的，如data:image/jpg;base64,
  • 图片格式：现支持PNG、JPG、JPEG、BMP，不支持GIF图片

package com.jzj.service;import com.jzj.utils.GsonUtils;
import com.jzj.utils.HttpUtil;import java.util.*;/**
* 人脸注册
*/
public class FaceAdd {/*** 重要提示代码中所需工具类* FileUtil,Base64Util,HttpUtil,GsonUtils请从* https://ai.baidu.com/file/658A35ABAB2D404FBF903F64D47C1F72* https://ai.baidu.com/file/C8D81F3301E24D2892968F09AE1AD6E2* https://ai.baidu.com/file/544D677F5D4E4F17B4122FBD60DB82B3* https://ai.baidu.com/file/470B3ACCA3FE43788B5A963BF0B625F3* 下载*/public static String add(String picture) {// 请求urlString url = "https://aip.baidubce.com/rest/2.0/face/v3/faceset/user/add";try {Map<String, Object> map = new HashMap<>();map.put("image", picture);// 用户组idmap.put("group_id", "test");// 用户id（由数字、字母、下划线组成），长度限制128Bmap.put("user_id", "user1");// 用户资料，长度限制256B 默认空map.put("user_info", "abc");// 活体检测控制 NORMAL: 一般的活体要求(平衡的攻击拒绝率, 通过率)map.put("liveness_control", "NORMAL");// 图片类型map.put("image_type", "BASE64");// 图片质量控制 LOW:较低的质量要求map.put("quality_control", "LOW");String param = GsonUtils.toJson(map);// 注意这里仅为了简化编码每一次请求都去获取access_token，线上环境access_token有过期时间， 客户端可自行缓存，过期后重新获取。String accessToken = AuthService.getAuth(); // 调用鉴权接口获取的tokenString result = HttpUtil.post(url, accessToken, "application/json", param);System.out.println(result);return result;} catch (Exception e) {e.printStackTrace();}return null;}
}

• 返回参数

{"face_token": "2fa64a88a9d5118916f9a303782a97d3", // 人脸图片的唯一标识// 人脸在图片中的位置"location": {"left": 117,"top": 131,"width": 172,"height": 170,"rotation": 4}
}

3.4 人脸搜索

也称为1：N识别，在指定人脸集合中，找到最相似的人脸

package com.jzj.service;import com.jzj.utils.GsonUtils;
import com.jzj.utils.HttpUtil;import java.util.*;/**
* 人脸搜索
*/
public class FaceSearch {/*** 重要提示代码中所需工具类* FileUtil,Base64Util,HttpUtil,GsonUtils请从* https://ai.baidu.com/file/658A35ABAB2D404FBF903F64D47C1F72* https://ai.baidu.com/file/C8D81F3301E24D2892968F09AE1AD6E2* https://ai.baidu.com/file/544D677F5D4E4F17B4122FBD60DB82B3* https://ai.baidu.com/file/470B3ACCA3FE43788B5A963BF0B625F3* 下载*/public static String faceSearch(String picture) {// 请求urlString url = "https://aip.baidubce.com/rest/2.0/face/v3/search";try {Map<String, Object> map = new HashMap<>();map.put("image", picture);map.put("liveness_control", "NORMAL");map.put("group_id_list", "test");map.put("image_type", "BASE64");map.put("quality_control", "LOW");String param = GsonUtils.toJson(map);// 注意这里仅为了简化编码每一次请求都去获取access_token，线上环境access_token有过期时间， 客户端可自行缓存，过期后重新获取。String accessToken = AuthService.getAuth();String result = HttpUtil.post(url, accessToken, "application/json", param);System.out.println(result);return result;} catch (Exception e) {e.printStackTrace();}return null;}
}

• 返回参数

{"face_token": "fid", // 人脸标志// 匹配的用户信息列表"user_list": [  {"group_id" : "test", // 用户所属的group_id"user_id": "u333333", // 用户的user_id"user_info": "Test User", // 注册用户时携带的user_info"score": 99.3  // 用户的匹配得分，推荐阈值80分}]
}

3.5 工具类

文件读取工具类

package com.jzj.utils;import java.io.*;/*** 文件读取工具类*/
public class FileUtil {/*** 读取文件内容，作为字符串返回*/public static String readFileAsString(String filePath) throws IOException {File file = new File(filePath);if (!file.exists()) {throw new FileNotFoundException(filePath);} if (file.length() > 1024 * 1024 * 1024) {throw new IOException("File is too large");} StringBuilder sb = new StringBuilder((int) (file.length()));// 创建字节输入流  FileInputStream fis = new FileInputStream(filePath);  // 创建一个长度为10240的Bufferbyte[] bbuf = new byte[10240];  // 用于保存实际读取的字节数  int hasRead = 0;  while ( (hasRead = fis.read(bbuf)) > 0 ) {  sb.append(new String(bbuf, 0, hasRead));  }  fis.close();  return sb.toString();}/*** 根据文件路径读取byte[] 数组*/public static byte[] readFileByBytes(String filePath) throws IOException {File file = new File(filePath);if (!file.exists()) {throw new FileNotFoundException(filePath);} else {ByteArrayOutputStream bos = new ByteArrayOutputStream((int) file.length());BufferedInputStream in = null;try {in = new BufferedInputStream(new FileInputStream(file));short bufSize = 1024;byte[] buffer = new byte[bufSize];int len1;while (-1 != (len1 = in.read(buffer, 0, bufSize))) {bos.write(buffer, 0, len1);}byte[] var7 = bos.toByteArray();return var7;} finally {try {if (in != null) {in.close();}} catch (IOException var14) {var14.printStackTrace();}bos.close();}}}
}

Base64 工具类

package com.jzj.utils;/*** Base64 工具类*/
public class Base64Util {private static final char last2byte = (char) Integer.parseInt("00000011", 2);private static final char last4byte = (char) Integer.parseInt("00001111", 2);private static final char last6byte = (char) Integer.parseInt("00111111", 2);private static final char lead6byte = (char) Integer.parseInt("11111100", 2);private static final char lead4byte = (char) Integer.parseInt("11110000", 2);private static final char lead2byte = (char) Integer.parseInt("11000000", 2);private static final char[] encodeTable = new char[]{'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '+', '/'};public Base64Util() {}public static String encode(byte[] from) {StringBuilder to = new StringBuilder((int) ((double) from.length * 1.34D) + 3);int num = 0;char currentByte = 0;int i;for (i = 0; i < from.length; ++i) {for (num %= 8; num < 8; num += 6) {switch (num) {case 0:currentByte = (char) (from[i] & lead6byte);currentByte = (char) (currentByte >>> 2);case 1:case 3:case 5:default:break;case 2:currentByte = (char) (from[i] & last6byte);break;case 4:currentByte = (char) (from[i] & last4byte);currentByte = (char) (currentByte << 2);if (i + 1 < from.length) {currentByte = (char) (currentByte | (from[i + 1] & lead2byte) >>> 6);}break;case 6:currentByte = (char) (from[i] & last2byte);currentByte = (char) (currentByte << 4);if (i + 1 < from.length) {currentByte = (char) (currentByte | (from[i + 1] & lead4byte) >>> 4);}}to.append(encodeTable[currentByte]);}}if (to.length() % 4 != 0) {for (i = 4 - to.length() % 4; i > 0; --i) {to.append("=");}}return to.toString();}
}

Json工具类

/** Copyright (C) 2017 Baidu, Inc. All Rights Reserved.*/
package com.jzj.utils;import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonParseException;import java.lang.reflect.Type;/*** Json工具类.*/
public class GsonUtils {private static Gson gson = new GsonBuilder().create();public static String toJson(Object value) {return gson.toJson(value);}public static <T> T fromJson(String json, Class<T> classOfT) throws JsonParseException {return gson.fromJson(json, classOfT);}public static <T> T fromJson(String json, Type typeOfT) throws JsonParseException {return (T) gson.fromJson(json, typeOfT);}
}

http 工具类

package com.jzj.utils;import java.io.BufferedReader;
import java.io.DataOutputStream;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.List;
import java.util.Map;/*** http 工具类*/
public class HttpUtil {public static String post(String requestUrl, String accessToken, String params)throws Exception {String contentType = "application/x-www-form-urlencoded";return HttpUtil.post(requestUrl, accessToken, contentType, params);}public static String post(String requestUrl, String accessToken, String contentType, String params)throws Exception {String encoding = "UTF-8";if (requestUrl.contains("nlp")) {encoding = "GBK";}return HttpUtil.post(requestUrl, accessToken, contentType, params, encoding);}public static String post(String requestUrl, String accessToken, String contentType, String params, String encoding)throws Exception {String url = requestUrl + "?access_token=" + accessToken;return HttpUtil.postGeneralUrl(url, contentType, params, encoding);}public static String postGeneralUrl(String generalUrl, String contentType, String params, String encoding)throws Exception {URL url = new URL(generalUrl);// 打开和URL之间的连接HttpURLConnection connection = (HttpURLConnection) url.openConnection();connection.setRequestMethod("POST");// 设置通用的请求属性connection.setRequestProperty("Content-Type", contentType);connection.setRequestProperty("Connection", "Keep-Alive");connection.setUseCaches(false);connection.setDoOutput(true);connection.setDoInput(true);// 得到请求的输出流对象DataOutputStream out = new DataOutputStream(connection.getOutputStream());out.write(params.getBytes(encoding));out.flush();out.close();// 建立实际的连接connection.connect();// 获取所有响应头字段Map<String, List<String>> headers = connection.getHeaderFields();// 遍历所有的响应头字段for (String key : headers.keySet()) {System.err.println(key + "--->" + headers.get(key));}// 定义 BufferedReader输入流来读取URL的响应BufferedReader in = null;in = new BufferedReader(new InputStreamReader(connection.getInputStream(), encoding));String result = "";String getLine;while ((getLine = in.readLine()) != null) {result += getLine;}in.close();System.err.println("result:" + result);return result;}
}

3.6 controller

package com.jzj.controller;import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.jzj.service.FaceAdd;
import com.jzj.service.FaceSearch;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;import javax.servlet.http.HttpServletRequest;/*** @author 黎明* @date 2023/7/12 9:38* @version 1.0*/
@RestController
@RequestMapping("/face")
public class FaceController {/*** 注册* @param request http请求* @return json*/@RequestMapping("/register")public String FaceRegister(HttpServletRequest request){System.out.println("---------faceregister-------");String image = request.getParameter("base");System.out.println(image);String result = FaceAdd.add(image);System.out.println("-----人脸注册成功");return result;}/*** 登录* @param request http请求* @return json*/@RequestMapping("/login")public String FaceLogin(HttpServletRequest request){String image = request.getParameter("base");String msg = FaceSearch.faceSearch(image);JSONObject json = JSONObject.parseObject(msg);JSONArray userList = json.getJSONObject("result").getJSONArray("user_list");if (userList != null && !userList.isEmpty()) {JSONObject userData = userList.getJSONObject(0);float score = userData.getFloatValue("score");if (score < 80){System.out.println("人脸匹配失败，请重新刷脸登录");return "";}System.out.println("相似度："+score);} else {System.out.println("未找到用户score数据");}return msg;}
}

3.7 前端页面

本次前端页面有用到jquery所以需要导入js文件

注册

DOCTYPE html>
<html lang="en">
<head><meta charset="UTF-8"><title>注册title>
head>
<body>
<dl class="admin_login"><dt><strong>strong><em>em> <strong>请把你的脸放摄像头面前strong>dt><div id="media"><video id="video" width="100%" height="300" autoplay>video><canvas id="canvas" width="530" height="300">canvas>div><dd><input type="button" onclick="query()" value="立即注册"class="submit_btn"/>dd>dl><script type="text/javascript" src="js/code.jquery.com_jquery-3.7.0.min.js">script>
<script type="text/javascript">//var 是定义变量var video = document.getElementById("video"); //获取video标签var context = canvas.getContext("2d");var con = {audio: false,video: true,video: {width: 1980,height: 1024,}};//导航 获取用户媒体对象navigator.mediaDevices.getUserMedia(con).then(function (stream) {try {video.src = window.URL.createObjectURL(stream);} catch (e) {video.srcObject = stream;}video.onloadmetadate = function (e) {video.play();}});function query() {//把流媒体数据画到convas画布上去context.drawImage(video, 0, 0, 530, 300);var base = getBase64();$.ajax({type: "post",url: "face/register",data: {"base": base},dataType: "json",success: function (json) {if (json != '') {$.MsgBox.Alert("消息", "注册成功");}}});}function getBase64() {var imgSrc = document.getElementById("canvas").toDataURL("image/png");return imgSrc.split("base64,")[1];};
script>
body>
html>

登录

DOCTYPE html>
<html lang="en">
<head><meta charset="UTF-8"><title>登录title>
head>
<body>
<dl class="admin_login"><dt><strong>strong><em>em> <strong>请把你的脸放摄像头面前strong>dt><div id="media"><video id="video" width="100%" height="300" autoplay>video><canvas id="canvas" width="530" height="300">canvas>div><dd><input type="button" onclick="query()" value="立即登录"class="submit_btn" />dd>
dl>
<script type="text/javascript" src="js/code.jquery.com_jquery-3.7.0.min.js">script>
<script type="text/javascript">//var 是定义变量var video = document.getElementById("video"); //获取video标签var context = canvas.getContext("2d");var con  ={audio:false,video:true,video:{width:1980,height:1024,}};//导航 获取用户媒体对象navigator.mediaDevices.getUserMedia(con).then(function(stream){try{video.src = window.URL.createObjectURL(stream);}catch(e){video.srcObject=stream;}video.onloadmetadate = function(e){video.play();}});function query(){//把流媒体数据画到convas画布上去context.drawImage(video,0,0,530,300);var base = getBase64();$.ajax({type:"post",url:"face/login",data:{"base":base},dataType: "json",success:function(json){if(json.message==""){alert("消息","人脸匹配失败,请重新刷脸登录");}else{window.parent.location.replace("welcome.html");}}});}function getBase64() {var imgSrc = document.getElementById("canvas").toDataURL("image/png");return imgSrc.split("base64,")[1];};
script>
body>
html>

登录成功欢迎页面

DOCTYPE html>
<html lang="en"><head><meta charset="UTF-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><title>Welcome to Our Websitetitle><style>body {font-family: Arial, sans-serif;text-align: center;background-color: #f2f2f2;}.welcome-message {margin-top: 200px;font-size: 24px;color: #333;}.user-details {margin-top: 50px;font-size: 18px;color: #777;}style>
head><body>
<h1 class="welcome-message">欢迎登录网站！h1>
<div class="user-details"><p>用户名：黎明p><p>邮箱：johndoe@example.comp>
div>
body>html>

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！