人脸识别登录demo源码

package org.web; import java.io.FileOutputStream; import java.io.IOException; import java.io.OutputStream; import java.io.PrintWriter; import java.util.Arrays; import java.util.HashMap; import javax.servlet.ServletException; import javax.servlet.http.HttpServlet; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import org.apache.struts2.ServletActionContext; import org.entity.Users; import org.json.JSONObject; import org.service.IUserService; import org.service.impl.UserServiceImpl; import sun.misc.BASE64Decoder; import com.baidu.aip.face.AipFace; public class FaceServlet extends HttpServlet { /** * (非 Javadoc) * <p> * Description(描述):get方法，主要调用post * </p> * <p> * Title: doGet * </p> * * @param request * @param response * @throws ServletException * @throws IOException * @see javax.servlet.http.HttpServlet#doGet(javax.servlet.http.HttpServletRequest, * javax.servlet.http.HttpServletResponse) */ public void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { doPost(request, response); } /** * (非 Javadoc) * <p> * Description(描述):post方法，主要进行业务操作 * </p> * <p> * Title: doPost * </p> * * @param request * @param response * @throws ServletException * @throws IOException * @see javax.servlet.http.HttpServlet#doPost(javax.servlet.http.HttpServletRequest, * javax.servlet.http.HttpServletResponse) */ public void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { // 防止乱码 response.setContentType("text/html;charset=utf-8"); request.setCharacterEncoding("utf-8"); response.setCharacterEncoding("utf-8"); String img = request.getParameter("img"); // 图像数据 String username = request.getParameter("username"); // 用户名 String tag = request.getParameter("tag"); String APP_ID = "11124682"; String API_KEY = "AGQszARPVp7h8vSWvpLfmC6E"; String SECRET_KEY = "G2s2VaLKlkauQeUbbxpQXDrob7X0fHBC"; AipFace client = new AipFace(APP_ID, API_KEY, SECRET_KEY); if (tag.equals("reg")) { // 注册 face(username, img, response, request, client); } else if (tag.endsWith("login")) { // 登陆 login(img, response, username, request, client); } } /** * * @Title: face * @Description: 该方法的主要作用：注册 1.将base64编码的图片保存 2.将图片路径保存在数据库里面 * 3.将人脸图片识别在人脸库中注册 * @param 设定文件 * @return 返回类型：void * @throws */ public void face(String username, String img, HttpServletResponse response, HttpServletRequest request, AipFace client) { Users user = new Users(); IUserService userService = new UserServiceImpl(); // 图片名称 String fileName = System.currentTimeMillis() + ".png"; String basePath = request.getSession().getServletContext() .getRealPath("picture/"); // 往数据库里面插入注册信息 user.setId(((Long) System.currentTimeMillis()).intValue()); user.setUsername(username); user.setHeadphoto("/picture/" + fileName); userService.save(user); // 往服务器里面上传图片 GenerateImage(img, basePath + "/" + fileName); // 给人脸库中加入一个脸 boolean flag = facesetAddUser(client, basePath + "/" + fileName, username); try { PrintWriter out = response.getWriter(); // 中文乱码，写个英文比较专业 哈哈 if (flag == false) { out.print("请把脸放上!!");// 请把两脸放上 } else { out.print("注册成功!!"); // 注册成功 } } catch (IOException e) { // TODO 异常执行块！ e.printStackTrace(); } } /** * * @Title: login * @Description: 该方法的主要作用：登陆 实现原理： 1.从前台接收用户名 2.然后把从前台传过来的图片先上传到服务器 * 3.把上传的这张人脸与人脸库中的人脸对应 * @param 设定文件 * @return 返回类型：String * @throws */ public void login(String img, HttpServletResponse response, String username, HttpServletRequest request, AipFace client) { // 图片名称 String fileName = System.currentTimeMillis() + ".png"; String basePath = request.getSession().getServletContext() .getRealPath("picture/"); // 往服务器里面上传图片 GenerateImage(img, basePath + "/" + fileName); response.setContentType("text/html,charset=utf-8;"); try { PrintWriter out = response.getWriter(); Double result = verifyUser(client, basePath + "/" + fileName, username); if (result > 90) { // 匹配成功 out.print("登陆成功"); } else { // 匹配失败 out.print("登陆失败"); } } catch (IOException e) { // TODO 异常执行块！ e.printStackTrace(); } } /** * * @Title: GenerateImage * @Description: 该方法的主要作用：// 对字节数组字符串进行Base64解码并生成图片 * @param @param imgStr * @param @param imgFilePath * @param @return 设定文件 * @return 返回类型：boolean * @throws */ public boolean GenerateImage(String imgStr, String imgFilePath) { if (imgStr == null) // 图像数据为空 return false; BASE64Decoder decoder = new BASE64Decoder(); try { // Base64解码 byte[] bytes = decoder.decodeBuffer(imgStr); for (int i = 0; i < bytes.length; ++i) { if (bytes[i] < 0) {// 调整异常数据 bytes[i] += 256; } } // 生成jpeg图片 OutputStream out = new FileOutputStream(imgFilePath); out.write(bytes); out.flush(); out.close(); return true; } catch (Exception e) { e.printStackTrace(); return false; } } /** * @Title: facesetAddUser * @Description: 该方法的主要作用：人脸注册,给人脸库中注册一个人脸 * @param @param client 设定文件 * @return 返回类型：void * @throws */ public boolean facesetAddUser(AipFace client, String path, String username) { // 参数为数据库中注册的人脸 HashMap<String, String> options = new HashMap<String, String>(); JSONObject res = client.addUser(username, "test_users_info", Arrays.asList("group1", "group2"), path, options); System.out.println(res.toString(2)); // {"log_id": 3621903690062422} if (res.keySet().contains("error_code")) { return false; } return true; } /** * * @Title: verifyUser * @Description: 该方法的主要作用：人脸认证 * @param @param client * @param @param path * @param @param username * @param @return 设定文件 * @return 返回类型：Double * @throws */ public Double verifyUser(AipFace client, String path, String username) { HashMap<String, Object> options = new HashMap<String, Object>(1); options.put("top_num", 5); /* * try { Thread.sleep(2000); } catch (InterruptedException e) { // TODO * 异常执行块！ e.printStackTrace(); } */ JSONObject res = client.verifyUser(username, Arrays.asList("group1", "group2"), path, options); Double result = Double.parseDouble(res.getJSONArray("result").get(0) .toString()); return result; } }