java实现人脸识别源码【含测试效果图】——Servlet层（FaceServlet）

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 = "9802974";
      String API_KEY = "OVYw5Ok0y9U8n6CfVPYt0wfZ";
      String SECRET_KEY = "aCN3lupCarq3rC9G8Rylqz1d36Towp8G";
      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;
  }
}