百度人脸识别API
https://ai.baidu.com/ai-doc/FACE/yk37c1u4t
获取access_token
由于百度API的access_token会定期更新，所以每次请求都会重新获得一个token
package com.example.emoswx.baiduFaceCheck; import org.json.JSONObject; import org.springframework.stereotype.Component; 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 = "1XbaXPvpQ4n66TjhpntXa0gy"; // 官网获取的 Secret Key 更新为您注册的 String clientSecret = "p5tbyEBb51pR9gRK1M9agipIpmAIrugI"; 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;
}
}
人脸检测
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.example.emoswx.baiduFaceCheck.utils.HttpUtil;
import com.example.emoswx.baiduFaceCheck.utils.GsonUtils;
import com.example.emoswx.exception.EmosException;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import java.util.*;
@Slf4j
@Component
public class BaiduFaceRecognition {
/** * 人脸检测与属性分析，是否人脸，是否模糊 */
public void faceDetect(String imgParam) {
// 请求url
String url = "https://aip.baidubce.com/rest/2.0/face/v3/detect";
try {
Map<String, Object> map = new HashMap<>();
map.put("max_face_num",1);
map.put("image_type", "BASE64");
map.put("image", imgParam);
map.put("face_field", "age,expression,face_shape,gender,glasses,quality,face_type");
String param = GsonUtils.toJson(map);
// 注意这里仅为了简化编码每一次请求都去获取access_token，线上环境access_token有过期时间， 客户端可自行缓存，过期后重新获取。
String accessToken = AuthService.getAuth();
String result = HttpUtil.post(url, accessToken, "application/json", param);
//result=JSONUtil.quote(result);
JSONObject resultJson= (JSONObject) JSONUtil.parseObj(result).get("result");
JSONObject faceListJson= JSONUtil.parseObj(JSONUtil.parseObj(resultJson, false, true).getJSONArray("face_list").get(0),false, true);
JSONObject qualityJson= (JSONObject) faceListJson.get("quality");
//判断是否成功
String msg= (String) JSONUtil.parseObj(result).get("error_msg");
if(!"SUCCESS".equals(msg)){
log.error(msg);
throw new EmosException(msg);
}
//照片的 face_token
//String face_token= (String) faceListJson.get("face_token");
//人脸模糊程度，范围[0~1]，0表示清晰，1表示模糊
int blur=qualityJson.getInt("blur");
//真实人脸置信度，[0~1]，大于0.5可以判断为人脸
int human= (int) faceListJson.get("face_probability");
//1表示人脸模糊
if(blur==1){
log.error("人脸模糊");
throw new EmosException("人脸模糊");
}
//是否是人类的脸
if(human==0){
log.error("提供的人脸照片非人类的脸");
throw new EmosException("提供的人脸照片非人类的脸");
}
// System.out.println("blur:"+blur);
// System.out.println("human:"+human);
// System.out.println(result);
} catch (Exception e) {
log.error(e.getMessage(),e);
throw new EmosException("人脸检测失败");
}
}
}
人脸注册
/** * 人脸信息入人脸库 * @return face_token * imgParam:人脸base64串 * 注册后会有生成一个face_token是永久的 * @throws Exception */
public String faceadd(String imgParam,int userId) throws Exception {
String url="https://aip.baidubce.com/rest/2.0/face/v3/faceset/user/add"; //人脸注册链接
Map<String, Object> map = new HashMap<>();
map.put("group_id", "face_group");
map.put("user_id", userId);
map.put("image_type", "BASE64");
map.put("image", imgParam);
map.put("quality_control", "LOW");
map.put("action_type","REPLACE");
String param = GsonUtils.toJson(map);
String accessToken = AuthService.getAuth();
try {
String result = HttpUtil.post(url, accessToken, "application/json",param);
//判断是否成功
String msg= (String) JSONUtil.parseObj(result).get("error_msg");
if(!"SUCCESS".equals(msg)){
log.error(msg);
throw new EmosException(msg);
}
JSONObject resultJson= (JSONObject) JSONUtil.parseObj(result).get("result");
return (String) resultJson.get("face_token");
}
catch (Exception e){
log.error(e.getMessage(),e);
throw new EmosException("人脸注册失败");
}
}
人脸识别 1：1
/** * 人脸信息识别，判断指定人脸是否存在库中 * @param imgParam:人脸信息，base64转码后的串 * @return 相似度 */
public double facerecognition(String imgParam,int userId) {
String url = "https://aip.baidubce.com/rest/2.0/face/v3/search";
try {
Map<String, Object> map = new HashMap<>();
map.put("image", imgParam);
map.put("user_id", userId);
map.put("group_id_list", "face_group");
map.put("image_type", "BASE64");
map.put("quality_control", "LOW");
String param = GsonUtils.toJson(map);
String accessToken = AuthService.getAuth();
String result = HttpUtil.post(url, accessToken, "application/json", param);
//人脸识别相似度
JSONObject resultJson= (JSONObject) JSONUtil.parseObj(result).get("result");
JSONObject userListJson= JSONUtil.parseObj(JSONUtil.parseObj(resultJson, false, true).getJSONArray("user_list").get(0),false, true);
double score=userListJson.getDouble("score");
//判断是否成功
String msg= (String) JSONUtil.parseObj(result).get("error_msg");
if(!"SUCCESS".equals(msg)){
log.error(msg);
throw new EmosException(msg);
}
if(score<85){
log.error("人脸不匹配");
throw new EmosException("人脸不匹配");
}
return score;
} catch (Exception e) {
log.error(e.getMessage(),e);
throw new EmosException(e.getMessage());
}
}
test
@Autowired
BaiduFaceRecognition baiduFaceRecognition;
@Test
void faceReconguration(){
//人脸检测 ok
try {
byte[] imgData =imgData = FileUtil.readFileByBytes("E:\\AA多啦A梦的兜\\message\\大头照.jpg");
String imgStr = Base64Util.encode(imgData);
baiduFaceRecognition.faceDetect(imgStr);
} catch (IOException e) {
e.printStackTrace();
}
//人脸注册 ok
try {
byte[] imgData =imgData = FileUtil.readFileByBytes("E:\\AA多啦A梦的兜\\message\\大头照.jpg");
String imgStr = Base64Util.encode(imgData);
baiduFaceRecognition.faceadd(imgStr,999);
} catch (Exception e) {
e.printStackTrace();
}
//人脸识别 ok
try {
byte[] imgData =imgData = FileUtil.readFileByBytes("E:\\AA多啦A梦的兜\\message\\白底.jpg");
String imgStr = Base64Util.encode(imgData);
System.out.println(baiduFaceRecognition.facerecognition(imgStr, 999));
} catch (Exception e) {
e.printStackTrace();
}
}