基于百度AI人脸识别技术的Demo

时间:2024-03-09 22:33:53

 

编写demo之前首先浏览官方API:http://ai.baidu.com/docs#/Face-API/top

下面是源码:

package com.examsafety.test;



import com.examsafety.test.baiduAi.Base64Util;
import com.examsafety.test.baiduAi.FileUtil;
import com.examsafety.test.baiduAi.HttpUtil;
import net.sf.json.JSONObject;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * 百度AI人脸识别demo
 *
 * @author Libin
 * @create 2017-12-20 16:35
 * @Copyright: www.sinotn.com. All rights reserved.
 **/
public class FaceAiDemo {
    /**
     * 百度云AK
     */
    private static final String API_KEY = "你的AK";
    /**
     * 百度云SK
     */
    private static final String SECRET_KEY = "你的SK";
    /**
     * 获取access_token的接口地址
     */
    private static final String AUTH_HOST = "https://aip.baidubce.com/oauth/2.0/token?";
    /**
     * 进行人脸探测的接口地址
     */
    private static final String DETECT_HOST = "https://aip.baidubce.com/rest/2.0/face/v1/detect";

    public static void main(String[] args){
        testDetect();
    }
    /**
     * 人脸探测调用方法
     */
    public static void testDetect(){
        /**
         * 照片路径集合:正式项目时可从数据库获取
         */
        List<String> filePathList = new ArrayList<String>();
        filePathList.add("C:\\Users\\Libin\\Desktop\\张三.jpg");
        filePathList.add("C:\\Users\\Libin\\Desktop\\李四.jpg");

        String imgFilePath = "";
        String expressionStr = "";
        String glassesStr = "";
        for (String imgPathStr : filePathList){
            JSONObject jsonObject = JSONObject.fromObject(detect(imgPathStr).optJSONArray("result").get(0));
            /**
             * 获取年龄、颜值分数、微笑程度、是否戴眼镜
             */
            double age = jsonObject.optDouble("age");
            double beauty = jsonObject.optDouble("beauty");
            int expression = jsonObject.optInt("expression");
            int glasses = jsonObject.optInt("glasses");

            switch (expression){
                case 0 : expressionStr = "不笑";
                    break;
                case 1 :  expressionStr = "微笑";
                    break;
                case 2 :  expressionStr = "大笑";
                    break;
                default: expressionStr = "无法识别";
            }

            switch (glasses){
                case 0 : glassesStr = "无眼镜";
                    break;
                case 1 :  glassesStr = "普通眼镜";
                    break;
                case 2 :  glassesStr = "墨镜";
                    break;
                default: glassesStr = "无法识别";
            }
            /**
             * 控制台打印输出探测结果
             * Tips:年龄默认为double,需用Math.round()四舍五入取整
             */
            System.out.println("年龄:" + Math.round(age));
            System.out.println("微笑程度:" + expressionStr);
            System.out.println("眼镜:" + glassesStr);
            System.out.println("颜值打分:" + beauty);

        }



    }

    /**
     * 重要提示代码中所需工具类
     * 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
     * 下载
     */
    /**
     * 人脸探测
     * @return
     */
    public static JSONObject detect(String filePath) {
        JSONObject jsonObject = null;
        try {
            byte[] imgData = FileUtil.readFileByBytes(filePath);
            String imgStr = Base64Util.encode(imgData);
            String imgParam = URLEncoder.encode(imgStr, "UTF-8");
            String param = "max_face_num=" + 5
                    + "&face_fields="
                    + "age,beauty,expression,faceshape,gender,glasses,landmark,race,qualities"
                    + "&image=" + imgParam;
            String accessToken = getAuth();
            String result = HttpUtil.post(DETECT_HOST, accessToken, param);
            jsonObject = JSONObject.fromObject(result);
            return jsonObject;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }




    /**
     * 获取权限token
     * @return
     */
    public static String getAuth(){
        // 获取token地址
        String getAccessTokenUrl = AUTH_HOST
                // 1. grant_type为固定参数
                + "grant_type=client_credentials"
                // 2. 官网获取的 API Key
                + "&client_id=" + API_KEY
                // 3. 官网获取的 Secret Key
                + "&client_secret=" + SECRET_KEY;
        JSONObject jsonObject = null;
        BufferedReader in = null;
        try {
            URL realUrl = new URL(getAccessTokenUrl);
            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的响应
            in = new BufferedReader(new InputStreamReader(connection.getInputStream()));
            String result = "";
            String line;
            while ((line = in.readLine()) != null) {
                result += line;
            }
            /**
             * 返回结果示例
             */
            jsonObject = JSONObject.fromObject(result);
            String access_token = jsonObject.getString("access_token");
            return access_token;
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if(in!=null){
                try {
                    in.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return null;
    }
}

 控制台打印: