百度AI人脸检测与对比

5.Spring Boot接口测试

package com.hz.controller;
import com.baidu.aip.face.AipFace;
import com.baidu.aip.face.MatchRequest;
import org.apache.tomcat.util.codec.binary.Base64;
import org.json.JSONObject;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;/**
* @Author: LNH
* @Date: 2024-11-13-9:45
* @Description:
*/
@RestController
public class FaceController {
    // 设置APPID/AK/SK
    private static final String APP_ID = "116216384";
    private static final String API_KEY = "VYnavoYcxNRGnahgk50WaTSl";
    private static final String SECRET_KEY = "qHhZBJIZSDfcrn830R05BkPyHoRLSqZ9";
    //人脸对比
    @PostMapping(value = "/match-faces", produces = "application/json")
    public ResponseEntity<String> matchFaces(@RequestParam("image1")
                                                MultipartFile image1,
                                               @RequestParam("image2")
                                                MultipartFile image2)
    throws IOException {
        // 初始化一个AipFace
        AipFace client = new AipFace(APP_ID, API_KEY, SECRET_KEY);
        // 读本地图片
        byte[] bytes1 = image1.getBytes();
        byte[] bytes2 = image2.getBytes();
        // 将字节转base64
        String imageStr1 = Base64.encodeBase64String(bytes1);
        String imageStr2 = Base64.encodeBase64String(bytes2);
        // 人脸对比
        MatchRequest req1 = new MatchRequest(imageStr1, "BASE64");
        MatchRequest req2 = new MatchRequest(imageStr2, "BASE64");
        ArrayList<MatchRequest> requests = new ArrayList<>();
        requests.add(req1);
        requests.add(req2);
        JSONObject res = client.match(requests);
        JSONObject result = handleMatchResult(res);
        // 返回 JSON 字符串
        return ResponseEntity.ok(result.toString());
}
//人脸检测
@PostMapping("/detect-face")
public ResponseEntity<String> detectFace(@RequestParam("image")
                                        MultipartFile image)
    throws
    IOException {
        // 初始化一个AipFace
        AipFace client = new AipFace(APP_ID, API_KEY, SECRET_KEY);
        // 读本地图片
        byte[] bytes = image.getBytes();
        // 将字节转base64
        String imageStr = Base64.encodeBase64String(bytes);
        // 人脸检测
        // 传入可选参数调用接口
        HashMap<String, String> options = new HashMap<>();
        options.put("face_field", "age");
        options.put("max_face_num", "2");
        options.put("face_type", "LIVE");
        options.put("liveness_control", "LOW");
        JSONObject res = client.detect(imageStr, "BASE64", options);
        JSONObject result = handleDetectResult(res);
        // 返回 JSON 字符串
        return ResponseEntity.ok(result.toString());
}
        //对比判断是否为同一个人
        private JSONObject handleMatchResult(JSONObject result) {
            int errorCode = result.getInt("error_code");
            if (errorCode == 0) { // 成功的情况
                double score = result.getJSONObject("result").getDouble("score");
                String isSamePerson = score > 80 ? "可能是同一个人" : "不是同一个人"; //
根据实际需求调整阈值
                result.put("isSamePerson", isSamePerson);
                result.put("score", score);
            } else {
                result.put("error_msg", "人脸对比失败: " +
                result.getString("error_msg"));
            }
                return result;
        }
        //人脸检测信息
        private JSONObject handleDetectResult(JSONObject result) {
            int errorCode = result.getInt("error_code");
            if (errorCode == 0) { // 成功的情况
                int faceNum = result.getJSONObject("result").getInt("face_num");
                if (faceNum > 0) {
                    JSONObject faceInfo =
                result.getJSONObject("result").getJSONArray("face_list").getJSONObject(0);
                    int age = faceInfo.getInt("age");
                } else {
                    result.put("error_msg", "人脸检测结果: 没有检测到人脸");
                }
            } else {
                result.put("error_msg", "人脸检测失败: " +
                result.getString("error_msg"));
            }
        return result;
    }
}
result.put("age", age);

<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>人脸对比与检测</title>
<style>
body {
font-family: Arial, sans-serif;
background-color: #f4f4f4;
margin: 0;
padding: 0;
}
h1 {
text-align: center;
color: #333;
}
form {
max-width: 600px;
margin: 20px auto;
padding: 20px;
background-color: #fff;
border-radius: 8px;
box-shadow: 0 0 10px rgba(0, 0, 0, 0.1);
}
input[type="file"] {
display: block;
margin-bottom: 10px;
}
button {
display: block;
width: 100%;
padding: 10px;
background-color: #007bff;
color: #fff;
border: none;
border-radius: 4px;
cursor: pointer;
font-size: 16px;
}
button:hover {
background-color: #0056b3;
}
#matchResult, #detectResult {
margin: 20px auto;
padding: 20px;
background-color: #fff;
border-radius: 8px;
box-shadow: 0 0 10px rgba(0, 0, 0, 0.1);
text-align: center;
font-size: 18px;
color: #333;
}
#matchResult p, #detectResult p {
margin: 0;
}
.success {
color: green;
}
.error {
color: red;
}
</style>
</head>
<body>
<h1>人脸对比与检测</h1>
<!-- 人脸对比表单 -->
<h2>人脸对比</h2>
<form id="matchForm">
<input type="file" name="image1" accept="image/*" required>
<input type="file" name="image2" accept="image/*" required>
<button type="submit">对比</button>
</form>
<div id="matchResult"></div>
<!-- 人脸检测表单 -->
<h2>人脸检测</h2>
<form id="detectForm">
<input type="file" name="image" accept="image/*" required>
<button type="submit">检测</button>
</form>
<div id="detectResult"></div>
<script src="https://code.jquery.com/jquery-3.7.1.min.js"></script>
<script>
$(document).ready(function() {
$('#matchForm').on('submit', function(event) {
event.preventDefault();
var formData = new FormData(this);
$.ajax({
url: '/match-faces',
type: 'POST',
data: formData,
contentType: false,
processData: false,
headers: {
'Accept': 'application/json'
},
success: function(response) {
console.log("Match response:", response);
if (response.isSamePerson && response.score !== undefined) {
$('#matchResult').html('<p class="success">人脸对比结果: '
+ response.isSamePerson + ' (相似度：' + response.score + ')</p>');
} else {
$('#matchResult').html('<p class="error">人脸对比失败: ' +
response.error_msg + '</p>');
}
},
error: function(error) {
console.error("Match error:", error);
$('#matchResult').html('<p class="error">人脸对比失败: ' +
error.responseJSON.error_msg + '</p>');
}
});
});
$('#detectForm').on('submit', function(event) {
event.preventDefault();
var formData = new FormData(this);
$.ajax({
url: '/detect-face',
type: 'POST',
data: formData,
contentType: false,
processData: false,
headers: {
'Accept': 'application/json'
},
success: function(response) {
console.log("Detect response:", response);
if (response.age !== undefined) {
$('#detectResult').html('<p class="success">人脸检测结果:
检测到人脸，年龄约为 ' + response.age + ' 岁</p>');
} else {
$('#detectResult').html('<p class="error">人脸检测结果: ' +
response.error_msg + '</p>');
}
},
error: function(error) {
console.error("Detect error:", error);
$('#detectResult').html('<p class="error">人脸检测失败: ' +
error.responseJSON.error_msg + '</p>');
}
});
});
});
</script>
</body>
</html>