From 5c750a5b35458d22a50206c22024a06d05956b9e Mon Sep 17 00:00:00 2001 From: yangjun <1173114630@qq.com> Date: Tue, 9 Jul 2024 16:36:55 +0800 Subject: [PATCH] =?UTF-8?q?=E6=AF=94=E5=AF=B9=E6=8E=A5=E5=8F=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/main/java/com/sqx/face/FaceMain.java | 181 +++++++++++++++++++++++ 1 file changed, 181 insertions(+) create mode 100644 src/main/java/com/sqx/face/FaceMain.java diff --git a/src/main/java/com/sqx/face/FaceMain.java b/src/main/java/com/sqx/face/FaceMain.java new file mode 100644 index 0000000..f3dad08 --- /dev/null +++ b/src/main/java/com/sqx/face/FaceMain.java @@ -0,0 +1,181 @@ +package com.sqx.face; + +import cn.hutool.core.thread.ThreadUtil; +import com.baidu.aip.face.AipFace; +import com.baidu.aip.face.MatchRequest; +import com.sqx.common.utils.RedisUtils; +import org.apache.commons.lang3.StringUtils; +import org.json.JSONObject; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.stereotype.Component; + +import java.io.File; +import java.util.ArrayList; + +@Component +public class FaceMain { + + @Autowired + private RedisUtils redisUtil; + + //设置APPID/AK/SK + public static final String APP_ID = "21388583"; + public static final String API_KEY = "K3bfbZRl7ttZp656n7Z8wSQCy"; + public static final String SECRET_KEY = "xEHt1sRGWlVtsNU10DiNhBMpWUqlTHNQ"; + public static final Long SLEEP_TIME = 1000L; + + + public AipFace creatApiClent(){ + // 可选：设置网络连接参数 +// client.setConnectionTimeoutInMillis(2000); +// client.setSocketTimeoutInMillis(60000); +// +// // 可选：设置代理服务器地址, http和socket二选一，或者均不设置 +// //client.setHttpProxy("proxy_host", proxy_port); // 设置http代理 +// //client.setSocketProxy("proxy_host", proxy_port); // 设置socket代理 +// +// // 调用接口 +// String image = "取决于image_type参数，传入BASE64字符串或URL字符串或FACE_TOKEN字符串"; +// String imageType = "BASE64"; +// +// // 人脸检测 +// JSONObject res = client.detect(image, imageType, options); +// System.out.println(res.toString(2)); + return new AipFace(APP_ID, API_KEY, SECRET_KEY); + } + + /** + * 人脸对比，需要传入URL(无系统地址) + * @param img1 对比的图片1 + * @param img2 对比的图片2 + * @return + */ + public JSONObject matchBySystemFileUrl(String img1, String img2) { + return matchBySystemFileUrl(img1, img2, true); + } + + /** + * 人脸对比，需要传入URL(无系统地址) + * @param img1 对比的图片1 + * @param img2 对比的图片2 + * @param isSleep 是否打开延时开关（true：加入延时，使用配置文件里的时间（单位：毫秒），false：不加延时） + * @return + */ + public JSONObject matchBySystemFileUrl(String img1, String img2, boolean isSleep) { + return matchBySystemFileUrl(img1, img2, isSleep, SLEEP_TIME); + } + + /** + * 人脸对比，需要传入URL(无系统地址) + * @param img1 对比的图片1 + * @param img2 对比的图片2 + * @param isSleep 是否打开延时开关（true：加入延时，false：不加延时） + * @param sleepTime 延时时间（单位：毫秒） + * @return + */ + public JSONObject matchBySystemFileUrl(String img1, String img2, boolean isSleep, Long sleepTime) { + if(StringUtils.isBlank(img1) || StringUtils.isBlank(img2)){ + return null; + } + //延迟一秒 + if(isSleep){ + ThreadUtil.sleep(1000); +// return null; + //ThreadUtil.sleep(sleepTime); + } + return match(img1,img2,"BASE64","BASE64"); + } + + /** + * 人脸对比，需要传入URL(无系统地址) + * @param img1 + * @param img2 + * @return + */ + public JSONObject matchBySystemFileUrl(String fillUrl, String img1, String img2) { + return match(fillUrl + File.separator + img1,fillUrl + File.separator + img2,"URL","URL"); + } + + /** + * 人脸对比，需要传入可访问的URL + * @param img1 + * @param img2 + * @return + */ + public JSONObject match(String img1,String img2) { + return match(img1,img2,"URL","URL"); + } + + public JSONObject match(String img1,String img2,String img1Type,String img2Type) { + // 初始化一个AipFace + AipFace client = creatApiClent(); + MatchRequest req1 = new MatchRequest(img1, img1Type); + MatchRequest req2 = new MatchRequest(img2, img2Type); + ArrayList requests = new ArrayList<>(); + requests.add(req1); + requests.add(req2); + + return client.match(requests); + } + + public static void main(String[] args) { + // 初始化一个AipFace + FaceMain faceMain = new FaceMain(); + AipFace client = faceMain.creatApiClent(); + + String image1 = "https://img0.baidu.com/it/u=2620353834,3334301303&fm=253&fmt=auto&app=138&f=JPEG?w=500&h=670"; + + String image2 = "https://img1.baidu.com/it/u=3388895316,506060607&fm=253&fmt=auto&app=138&f=JPEG?w=608&h=338"; + + MatchRequest req1 = new MatchRequest(image1, "URL");// + MatchRequest req2 = new MatchRequest(image2, "URL"); + ArrayList requests = new ArrayList<>(); + requests.add(req1); + requests.add(req2); + + JSONObject res = client.match(requests); + System.out.println(res.toString(2)); + } + + public void addRunSize(Integer currentNum, Integer allNum){ + addRunSize(currentNum); + addRunAllSize(allNum); + } + //正在执行数 + public void addRunSize(Integer num) { + redisUtil.set("calcAllFaceDataIsRunSize",num,SLEEP_TIME * 60 * 5); + } + + //执行总数 + public void addRunAllSize(Integer num) { + redisUtil.set("calcAllFaceDataIsRunAllSize",num,SLEEP_TIME * 60 * 5); + } + + public String getPromptStr(){ + Integer calcAllFaceDataIsRunSize = Integer.parseInt(redisUtil.get("calcAllFaceDataIsRunSize")); + Integer calcAllFaceDataIsRunAllSize = Integer.parseInt(redisUtil.get("calcAllFaceDataIsRunAllSize")); + calcAllFaceDataIsRunSize = calcAllFaceDataIsRunSize == null?0:calcAllFaceDataIsRunSize; + calcAllFaceDataIsRunAllSize = calcAllFaceDataIsRunAllSize == null?0:calcAllFaceDataIsRunAllSize; + return calcAllFaceDataIsRunSize + "/" + calcAllFaceDataIsRunAllSize; + } + + public boolean isRun(){ + Boolean isRun = Boolean.parseBoolean(redisUtil.get("calcAllFaceDataIsRun")); + if(isRun != null && isRun){ + //redisUtil.del("calcAllFaceDataIsRun"); + return true; + } + return false; + } + + public void runFlag() { + redisUtil.set("calcAllFaceDataIsRun",true,SLEEP_TIME * 60 * 5); + } + + public void delRunFlag() { + redisUtil.delete("calcAllFaceDataIsRunSize"); + redisUtil.delete("calcAllFaceDataIsRunAllSize"); + redisUtil.delete("calcAllFaceDataIsRun"); + } +}