首先去百度ai开通人脸对比接口应用才有接下来的故事
地址:https://ai.baidu.com/tech/face/compare
1.导入人脸对比SDK,我使用maven自动下载,不嫌麻烦可以下载再人工导入https://ai.baidu.com/sdk#bfr
<dependency>
<groupId>com.baidu.aip</groupId>
<artifactId>java-sdk</artifactId>
<version>4.12.0</version>
</dependency>
2.下载所需工具类
FileUtil,HttpUtil,GsonUtils
https://ai.baidu.com/file/658A35ABAB2D404FBF903F64D47C1F72
https://ai.baidu.com/file/544D677F5D4E4F17B4122FBD60DB82B3
https://ai.baidu.com/file/470B3ACCA3FE43788B5A963BF0B625F3
其中GsonUtils可以使用maven下载
<dependency>
<groupId>com.google.code.gson</groupId>
<artifactId>gson</artifactId>
<version>2.8.5</version>
</dependency>
3.按照官方文档获取access_token
文档地址:https://cloud.baidu.com/doc/FACE/s/Lk37c1tpf
方便懒人直接给出代码:
import org.json.JSONObject;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.List;
import java.util.Map;
/**
* 获取token类
*/
public class AuthService {
/**
* 获取权限token
* @return 返回示例:
* {
* "access_token": "24.460da4889caad24cccdb1fea17221975.2592000.1491995545.282335-1234567",
* "expires_in": 2592000
* }
*/
public static String getAuth() {
// 官网获取的 API Key 更新为你注册的
String clientId = "百度云应用的AK";
// 官网获取的 Secret Key 更新为你注册的
String clientSecret = "百度云应用的SK";
return getAuth(clientId, clientSecret);
}
/**
* 获取API访问token
* 该token有一定的有效期,需要自行管理,当失效时需重新获取.
* @param ak - 百度云官网获取的 API Key
* @param sk - 百度云官网获取的 Securet Key
* @return assess_token 示例:
* "24.460da4889caad24cccdb1fea17221975.2592000.1491995545.282335-1234567"
*/
public static String getAuth(String ak, String sk) {
// 获取token地址
String authHost = "https://aip.baidubce.com/oauth/2.0/token?";
String getAccessTokenUrl = authHost
// 1. grant_type为固定参数
+ "grant_type=client_credentials"
// 2. 官网获取的 API Key
+ "&client_id=" + ak
// 3. 官网获取的 Secret Key
+ "&client_secret=" + sk;
try {
URL realUrl = new URL(getAccessTokenUrl);
// 打开和URL之间的连接
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的响应
BufferedReader in = new BufferedReader(new InputStreamReader(connection.getInputStream()));
String result = "";
String line;
while ((line = in.readLine()) != null) {
result += line;
}
/**
* 返回结果示例
*/
System.err.println("result:" + result);
JSONObject jsonObject = new JSONObject(result);
String access_token = jsonObject.getString("access_token");
return access_token;
} catch (Exception e) {
System.err.printf("获取token失败!");
e.printStackTrace(System.err);
}
return null;
}
}
4.编写测试类Test.java
package com.smarthome;
import com.baidu.aip.util.Base64Util;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import com.smarthome.util.FileUtil;
import com.smarthome.util.GetAccessToken;
import com.smarthome.util.GsonUtils;
import com.smarthome.util.HttpUtil;
import java.util.*;
public class Test {
public static void main(String[] args) throws Exception {
String result=match();
JsonObject jObject = new JsonParser().parse(result).getAsJsonObject();
JsonElement jsonElement = jObject.get("result");
float score = 0;
if(!jsonElement.isJsonNull()) {
score = jsonElement.getAsJsonObject().get("score").getAsFloat();
}
System.out.println("score is "+score);
}
public static String match() throws Exception {
//GetAccessToken.getAuth()获取access_token,30天过期
//人脸对比接口
//String url = "https://aip.baidubce.com/rest/2.0/face/v3/match?access_token=" + GetAccessToken.getAuth() + "";
byte[] bytes1 = FileUtil.readFileByBytes("/Users/zhangtaiqing/Desktop/1.PNG");
byte[] bytes2 = FileUtil.readFileByBytes("/Users/zhangtaiqing/Desktop/3.JPG");
String image1 = Base64Util.encode(bytes1);
String image2 = Base64Util.encode(bytes2);
List<Map<String, Object>> images = new ArrayList<>();
//请求参数详情https://ai.baidu.com/ai-doc/FACE/8k37c1rqz#人脸对比
Map<String, Object> map1 = new HashMap<>();
map1.put("image", image1);
map1.put("image_type", "BASE64");
map1.put("face_type", "IDCARD");
map1.put("quality_control", "LOW");
map1.put("liveness_control", "NORMAL");
Map<String, Object> map2 = new HashMap<>();
map2.put("image", image2);
map2.put("image_type", "BASE64");
map2.put("face_type", "LIVE");
map2.put("quality_control", "LOW");
map2.put("liveness_control", "NORMAL");
images.add(map1);
images.add(map2);
String param = GsonUtils.toJson(images);
// 注意这里仅为了简化编码每一次请求都去获取access_token,线上环境access_token有过期时间, 客户端可自行缓存,过期后重新获取。
String accessToken = GetAccessToken.getAuth();
String result = HttpUtil.post("https://aip.baidubce.com/rest/2.0/face/v3/match", accessToken, "application/json", param);
return result;
}
}