基於百度AI人臉識別技術的Demo

 

編寫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;
    }
}

 控制台打印：