人脸识别demo

This commit is contained in:
dengwenjie
2025-02-23 12:56:50 +08:00
parent b8f7b871be
commit 173e3f1e63
6 changed files with 490 additions and 14 deletions

View File

@@ -16,6 +16,7 @@
<exec.mainClass>ai.djl.examples.inference.cv.ObjectDetection</exec.mainClass>
</properties>
<dependencies>
<dependency>
<groupId>commons-cli</groupId>
@@ -42,7 +43,7 @@
<dependency>
<groupId>ink.numberone</groupId>
<artifactId>smartjavaai-face</artifactId>
<version>1.0.0</version>
<version>1.0.1</version>
</dependency>
<dependency>

View File

@@ -32,7 +32,7 @@ public class FaceDemo {
public static void main(String[] args) {
try {
//detectFace();
detectFace();
//verifyIDCard();
} catch (Exception e) {
e.printStackTrace();
@@ -52,9 +52,9 @@ public class FaceDemo {
FaceDetectedResult result = currentAlgorithm.detect("src/main/resources/largest_selfie.jpg");
logger.info("人脸检测结果:{}", JSONObject.toJSONString(result));
//使用图片流检测
//File imageFile = new File("/Users/wenjie/Downloads/djl-master/examples/src/test/resources/largest_selfie.jpg");
//FaceDetectedResult result = currentAlgorithm.detect(new FileInputStream(imageFile));
File input = new File("src/main/resources/largest_selfie.jpg");
//File input = new File("src/main/resources/largest_selfie.jpg");
//FaceDetectedResult result = currentAlgorithm.detect(new FileInputStream(input));
//logger.info("人脸检测结果:{}", JSONObject.toJSONString(result));
BufferedImage image = ImageIO.read(input);
//创建保存路径
Path imagePath = Paths.get("output").resolve("retinaface_detected.jpg");
@@ -96,8 +96,8 @@ public class FaceDemo {
//提取身份证人脸特征(图片仅供测试)
float[] featureIdCard = currentAlgorithm.featureExtraction("src/main/resources/kana1.jpg");
//提取身份证人脸特征(从图片流获取)
//File imageFile = new File("/Users/wenjie/Downloads/djl-master/examples/src/test/resources/largest_selfie.jpg");
//float[] featureIdCard = currentAlgorithm.featureExtraction(new FileInputStream(imageFile));
//File input = new File("src/main/resources/kana1.jpg");
//float[] featureIdCard = currentAlgorithm.featureExtraction(new FileInputStream(input));
logger.info("身份证人脸特征:{}", JSONObject.toJSONString(featureIdCard));
//提取实时人脸特征(图片仅供测试)
float[] realTimeFeature = currentAlgorithm.featureExtraction("src/main/resources/kana2.jpg");