1. 程式人生 > >基於百度AI人臉識別技術的Demo

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

-a import arr readfile red ets 2.0 路徑 之前

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

控制臺打印:

技術分享圖片

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