人脸识别

liujun001
1 parent 0ac09b45
Showing 10 changed files with 185 additions and 15 deletions
Bsth-admin/pom.xml
Bsth-admin/src/main/java/com/ruoyi/config/BsthSystemConfig.java
Bsth-admin/src/main/java/com/ruoyi/config/FaceEngineConfig.java
Bsth-admin/src/main/java/com/ruoyi/domain/driver/NewDriver.java
Bsth-admin/src/main/java/com/ruoyi/service/impl/dss/FaceServiceImpl.java
Bsth-admin/src/main/resources/application-druid-dev.yml
Bsth-admin/src/main/resources/libs/WIN64/libarcsoft_face.dll
Bsth-admin/src/main/resources/libs/WIN64/libarcsoft_face_engine.dll
Bsth-admin/src/main/resources/libs/WIN64/libarcsoft_face_engine_jni.dll
Bsth-admin/src/main/resources/libs/arcsoft-sdk-face-3.0.0.0.jar
@@ -68,7 +68,7 @@
             <version>1.6.2</version>
         </dependency>
-         <!-- Mysql驱动包 -->
+        <!-- Mysql驱动包 -->
         <dependency>
             <groupId>com.mysql</groupId>
             <artifactId>mysql-connector-j</artifactId>
@@ -113,6 +113,12 @@
             <version>4.5.13</version>
         </dependency>
+        <dependency>
+            <groupId>com.arcsoft.face</groupId>
+            <artifactId>arcsoft-sdk-face</artifactId>
+            <version>3.0.0</version>
+        </dependency>
+
     </dependencies>
     <build>
@@ -132,15 +138,15 @@
                     </execution>
                 </executions>
             </plugin>
-            <plugin>   
-                <groupId>org.apache.maven.plugins</groupId>   
-                <artifactId>maven-war-plugin</artifactId>   
-                <version>3.1.0</version>   
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-war-plugin</artifactId>
+                <version>3.1.0</version>
                 <configuration>
                     <failOnMissingWebXml>false</failOnMissingWebXml>
                     <warName>${project.artifactId}</warName>
-                </configuration>   
-           </plugin>   
+                </configuration>
+            </plugin>
         </plugins>
         <finalName>${project.artifactId}-sign</finalName>
     </build>
@@ -16,4 +16,12 @@ public class BsthSystemConfig {
     @Value("${bsth.upload.video.basePath}")
     private String videoBasePath;
+
+    @Value("${bsth.face.app.id}")
+    private String faceAppId;
+    @Value("${bsth.face.sdk.key}")
+    private String faceSdkKey;
+    @Value("${bsth.face.lib.path}")
+    private String faceLibPath;
+
 }
+package com.ruoyi.config;
+
+import com.arcsoft.face.ActiveFileInfo;
+import com.arcsoft.face.EngineConfiguration;
+import com.arcsoft.face.FaceEngine;
+import com.arcsoft.face.FunctionConfiguration;
+import com.arcsoft.face.enums.DetectMode;
+import com.arcsoft.face.enums.DetectOrient;
+import com.arcsoft.face.enums.ErrorInfo;
+import com.ruoyi.exception.InterruptException;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.context.annotation.Bean;
+import org.springframework.stereotype.Component;
+
+import javax.annotation.PreDestroy;
+import java.util.Objects;
+
+/**
+ * @author liujun
+ * @date 2024年07月25日 16:23
+ */
+@Slf4j
+@Component
+public class FaceEngineConfig {
+    private FaceEngine faceEngine;
+
+    @Bean
+    public FaceEngine createFaceEngine(BsthSystemConfig bsthSystemConfig) {
+        FaceEngine faceEngine = new FaceEngine(bsthSystemConfig.getFaceLibPath());
+//        FaceEngine faceEngine = new FaceEngine("D:\\\\arcsoft_lib");
+        //激活引擎
+        int errorCode = faceEngine.activeOnline(bsthSystemConfig.getFaceAppId(), bsthSystemConfig.getFaceSdkKey());
+
+        if (errorCode != ErrorInfo.MOK.getValue() && errorCode != ErrorInfo.MERR_ASF_ALREADY_ACTIVATED.getValue()) {
+            throw new InterruptException("引擎激活失败");
+        }
+
+
+        ActiveFileInfo activeFileInfo = new ActiveFileInfo();
+        errorCode = faceEngine.getActiveFileInfo(activeFileInfo);
+        if (errorCode != ErrorInfo.MOK.getValue() && errorCode != ErrorInfo.MERR_ASF_ALREADY_ACTIVATED.getValue()) {
+            throw new InterruptException("获取激活文件信息失败");
+        }
+
+        //引擎配置
+        EngineConfiguration engineConfiguration = new EngineConfiguration();
+        engineConfiguration.setDetectMode(DetectMode.ASF_DETECT_MODE_IMAGE);
+        engineConfiguration.setDetectFaceOrientPriority(DetectOrient.ASF_OP_ALL_OUT);
+        engineConfiguration.setDetectFaceMaxNum(10);
+        engineConfiguration.setDetectFaceScaleVal(16);
+        //功能配置
+        FunctionConfiguration functionConfiguration = new FunctionConfiguration();
+        functionConfiguration.setSupportAge(true);
+        functionConfiguration.setSupportFace3dAngle(true);
+        functionConfiguration.setSupportFaceDetect(true);
+        functionConfiguration.setSupportFaceRecognition(true);
+        functionConfiguration.setSupportGender(true);
+        functionConfiguration.setSupportLiveness(true);
+        functionConfiguration.setSupportIRLiveness(true);
+        engineConfiguration.setFunctionConfiguration(functionConfiguration);
+
+
+        //初始化引擎
+        errorCode = faceEngine.init(engineConfiguration);
+
+        if (errorCode != ErrorInfo.MOK.getValue()) {
+            throw new InterruptException("初始化引擎失败");
+        }
+
+        return faceEngine;
+    }
+
+    @PreDestroy
+    public void faceEngineConfigPreDestory() {
+        if (Objects.nonNull(faceEngine)) {
+            faceEngine.unInit();
+        }
+    }
+}
 package com.ruoyi.domain.driver;
-import lombok.Data;
 import com.baomidou.mybatisplus.annotation.IdType;
 import com.baomidou.mybatisplus.annotation.TableField;
 import com.baomidou.mybatisplus.annotation.TableId;
 import com.baomidou.mybatisplus.annotation.TableName;
-import org.apache.commons.lang3.StringUtils;
+import com.ruoyi.common.annotation.Excel;
+import lombok.Data;
 import lombok.EqualsAndHashCode;
 import lombok.experimental.Accessors;
 import lombok.extern.slf4j.Slf4j;
-import com.ruoyi.common.annotation.Excel;
 @Data
 @Slf4j
@@ -112,6 +111,9 @@ public class NewDriver {
     @Excel(name = "头像")
     private String image;
+    @TableField(exist = false)
+    private float imageScore;
+
     /***更新日期*/
     @Excel(name = "更新日期")
 package com.ruoyi.service.impl.dss;
+import cn.hutool.core.codec.Base64;
+import com.arcsoft.face.FaceEngine;
+import com.arcsoft.face.FaceFeature;
+import com.arcsoft.face.FaceInfo;
+import com.arcsoft.face.FaceSimilar;
+import com.arcsoft.face.enums.ErrorInfo;
+import com.arcsoft.face.toolkit.ImageInfo;
+import com.ruoyi.config.BsthSystemConfig;
 import com.ruoyi.domain.driver.NewDriver;
 import com.ruoyi.service.driver.NewDriverService;
 import com.ruoyi.service.dss.FaceService;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.collections4.CollectionUtils;
+import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
-import java.util.List;
+import java.io.File;
+import java.util.*;
+
+import static com.arcsoft.face.toolkit.ImageFactory.getRGBData;
 /**
  * @author liujun
@@ -19,18 +31,74 @@ import java.util.List;
 public class FaceServiceImpl implements FaceService {
     @Autowired
     private NewDriverService newDriverService;
+    @Autowired
+    private FaceEngine faceEngine;
+    @Autowired
+    private BsthSystemConfig bsthSystemConfig;
+
     @Override
     public NewDriver checkFace(NewDriver driver) {
         List<NewDriver> drivers = newDriverService.list(new NewDriver());
-        return checkFace(drivers);
+        return checkFace(drivers, driver);
     }
-    private NewDriver checkFace(List<NewDriver> drivers) {
+    private NewDriver checkFace(List<NewDriver> drivers, NewDriver sourceDriver) {
         if (CollectionUtils.isEmpty(drivers)) {
             return null;
         }
         return null;
     }
+
+    private float checkFaceScore(NewDriver dr, FaceFeature sourceFaceFeature) {
+        if (StringUtils.isEmpty(dr.getImage())) {
+            log.error("头像路径为空;[{}]", dr);
+            return -1;
+        }
+        StringBuilder builder = new StringBuilder();
+        builder.append(bsthSystemConfig.getImageBasePath());
+        builder.append(File.separator);
+        builder.append(dr.getImage());
+
+        File targetFile = new File(builder.toString());
+        if (!targetFile.exists() || !targetFile.isFile()) {
+            log.error("没有比对的图片;[{}]", dr);
+            return -1;
+        }
+
+        ImageInfo target = getRGBData(targetFile);
+        FaceFeature targetFaceFeature = generateFaceFeature(dr, target);
+        if (Objects.isNull(targetFaceFeature)) {
+            return -1;
+        }
+
+        //特征比对
+        FaceSimilar faceSimilar = new FaceSimilar();
+        int errorCode = faceEngine.compareFaceFeature(targetFaceFeature, sourceFaceFeature, faceSimilar);
+
+        if (errorCode != ErrorInfo.MOK.getValue()) {
+            log.error("人脸对比失败:[{}]", dr);
+            return -1;
+        }
+        return faceSimilar.getScore();
+    }
+
+    private FaceFeature generateFaceFeature(NewDriver driver, ImageInfo image) {
+        List<FaceInfo> faceInfoList = new ArrayList<>();
+        int errorCode = faceEngine.detectFaces(image.getImageData(), image.getWidth(), image.getHeight(), image.getImageFormat(), faceInfoList);
+        if (errorCode != ErrorInfo.MOK.getValue()) {
+            log.error("人脸对比失败，请检查数据:[{}]", driver);
+            return null;
+        }
+
+        FaceFeature faceFeature = new FaceFeature();
+        errorCode = faceEngine.extractFaceFeature(image.getImageData(), image.getWidth(), image.getHeight(), image.getImageFormat(), faceInfoList.get(0), faceFeature);
+        if (errorCode != ErrorInfo.MOK.getValue()) {
+            log.error("提取对比特征失败，请检查数据:[{}]", driver);
+            return null;
+        }
+        return faceFeature;
+    }
+
 }
@@ -199,5 +199,12 @@ bsth:
     upload:
         image:
             basePath: D:/temp/temp/bsth/upload
-        vide:
-            basePath: D:/temp/temp/bsth/upload
 \ No newline at end of file
+        video:
+            basePath: D:/temp/temp/bsth/upload
+    face:
+        app:
+            id: 8jPk3SNnaoGsd9SidMefgZXg1zbst64jB44vVyx9Cijq
+        sdk:
+            key: C21s5J1n1rHwXPkvVjubKshtofV5sHXvyUQqSWYxHp2b
+        lib:
+            path: D:/work/code/jienengjiancha/bsth-alcohol-sign/Bsth-admin/src/main/resources/libs/WIN64
 \ No newline at end of file