zoukankan      html  css  js  c++  java
  • 虹软人脸识别-SpringBoot集成

    一、前言

    ​ 人工智能时代的到来,相信大家已耳濡目染,虹软免费离线开放的人脸识别 SDK,正推动着全行业进入刷脸时代。为了方便开发者接入,虹软提供了多种语言,多种平台的人脸识别SDK的支持,使用场景广泛。产品主要功能有:人脸检测、追踪、特征提取、特征比对、属性检测,活体检测,图像质量检测等。此外,虹软提供的是基于本地算法特征的离线识别SDK,提供全平台的离线支持。

    ​ 作为一名刚接触人脸识别的初学者,对于虹软极为简洁,方便的SDK接入充满了好奇,想试图应用到web领域,而如今Java最火的web框架非SpringBoot莫属。但对于Java语言,虹软官网暂时还没有提供基于SpringBoot的集成Demo,因此便尝试写个将Java的人脸识别SDK和SpringBoot进行集成的样例,并写此文章进行记录,向广大初学开发者作分享。

    ​ 此Demo采用Maven作为项目管理工具,并基于Windows x64,Java 8 以及 SpringBoot 2.1.6,SDK是基于虹软人脸识别 SDK3.0。

    二、项目结构

    ​ SDK依赖Jar包 可从虹软官网获取 点击”免费获取” , ”登录“后 选择 具体“平台/版本/语言“ 获取。

    三、项目依赖

    ​ pom.xml 依赖包括

    • SpringBoot-Web依赖
    • SpringBoot-Devtools热部署依赖
    • SpringBoot-Freemarker依赖,
    • Hutool,Fastjson, Lombok,Commons-pool2,Guava
    • 虹软人脸识别SDK依赖Jar包
    • SpringBoot-Maven插件
    <dependency>
        <groupId>org.projectlombok</groupId>
        <artifactId>lombok</artifactId>
        <version>1.18.8</version>
    </dependency>
    
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
    
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-devtools</artifactId>
        <optional>true</optional>
    </dependency>
    
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-freemarker</artifactId>
    </dependency>
    
    <dependency>
        <groupId>cn.hutool</groupId>
        <artifactId>hutool-all</artifactId>
        <version>4.6.1</version>
    </dependency>
    
    <dependency>
        <groupId>com.alibaba</groupId>
        <artifactId>fastjson</artifactId>
        <version>1.2.59</version>
    </dependency>
    
    <dependency>
        <groupId>org.apache.commons</groupId>
        <artifactId>commons-pool2</artifactId>
        <version>2.6.0</version>
    </dependency>
    
    <dependency>
        <groupId>com.google.guava</groupId>
        <artifactId>guava</artifactId>
        <version>26.0-jre</version>
    </dependency>
    
    <dependency>
        <groupId>com.arcsoft.face</groupId>
        <artifactId>arcsoft-sdk-face</artifactId>
        <version>3.0.0.0</version>
        <scope>system</scope>
        <systemPath>${basedir}/lib/arcsoft-sdk-face-3.0.0.0.jar</systemPath>
    </dependency>
     <build>
            <plugins>
                <plugin>
                    <groupId>org.springframework.boot</groupId>
                    <artifactId>spring-boot-maven-plugin</artifactId>
                    <configuration>
                        <includeSystemScope>true</includeSystemScope>
                        <fork>true</fork>
                    </configuration>
                </plugin>
            </plugins>
        </build>
    

    四、项目流程

    五、效果展示

    ​ 在Application启动类 右击 选择 Run Application 即可运行程序,待程序启动完成后,访问 http://127.0.0.1:8089/

    六、核心代码说明

    1. application.properties 配置说明
    #上传文件 最大值限制
    spring.servlet.multipart.max-file-size=100MB
    #请求 最大值限制
    spring.servlet.multipart.max-request-size=100MB
    #请求头 最大值限制
    server.max-http-header-size=2MB
    #请求体 最大值限制
    server.tomcat.max-http-post-size=50MB
    #项目访问端口
    server.port=8089
    #人脸识别引擎库路径
    config.arcface-sdk.sdk-lib-path=d:/arcsoft_lib
    #sdk appId
    config.arcface-sdk.app-id=9iSfMeAhj********************Kes2TpSrd
    #sdk sdkKey
    config.arcface-sdk.sdk-key=BuRTH3hGs9*******************yP9xu6fiFG7G
    #人脸识别 引擎池大小
    config.arcface-sdk.detect-pool-size=5
    #人脸比对 引擎池大小
    config.arcface-sdk.compare-pool-size=5
    #关闭freemarker模板引擎缓存
    spring.freemarker.cache=false
    #模板引擎更新延迟设置为0
    spring.freemarker.settings.template_update_delay=0
    

    ​ 其中 人脸识别引擎库,APP_ID,SDK_KEY 可通过虹软官网”开发者中心“,进行 “登录”后 在“我的应用“中进行获取。

    2. 项目实体类说明

    1)UserRamCache 人脸信息存储类

    public class UserRamCache {
        private static  ConcurrentHashMap<String, UserInfo> userInfoMap = new ConcurrentHashMap<>();
        public static void addUser(UserInfo userInfo) {
            userInfoMap.put(userInfo.getFaceId(), userInfo);
        }
        public static void removeUser(String faceId) {
            userInfoMap.remove(faceId);
        }
        public static List<UserInfo> getUserList() {
            List<UserInfo> userInfoList = Lists.newLinkedList();
            for (UserInfo value : userInfoMap.values()) {
                userInfoList.add(value);
            }
            return userInfoList;
        }
        @Data
        public static class UserInfo {
            //人脸Id
            private String faceId;
            //人脸名称
            private String name;
            //人脸特征值
            private byte[] faceFeature;
        }
    }
    

    ​ 此类拥有一个 UserInfo的内部类,用于封装人脸信息,userInfoMap以人脸名称为key,UserInfo对象为Value 存储 并提供相应增/删/查功能的方法。

    2)ProcessInfo 人脸检测实体类

    public class ProcessInfo {
        //年龄
        private int age;
        //性别
        private int gender;
        //是否活体
        private int liveness;
    }
    

    3)UserCompareInfo 人脸识别实体类 此类继承自 人脸信息存储类的人脸信息类(内部类)

    public class UserCompareInfo extends UserRamCache.UserInfo {
        //人脸比对后的相似值
        private Float similar;
    }
    

    4)FaceDetectResDTO 人脸检测DTO封装类

    public class FaceDetectResDTO {
        //人脸框
        private Rect rect;
        //人脸角度
        private int orient;
        //人脸Id
        private int faceId = -1;
        //年龄
        private int age = -1;
        //性别
        private int gender = -1;
        //是否为活体
        private int liveness = -1;
    }
    

    5)FaceRecognitionResDTO 人脸识别DTO封装类

    public class FaceRecognitionResDTO {
        //人脸框
        private Rect rect;
        //人脸名称
        private String name;
        //人脸比对相似值
        private float similar;
    }
    
    3. FaceEngineFactory类 源码说明

    ​ 此类继承自BasePooledObjectFactory抽象类,为FaceEngine对象池。

    1)成员变量说明

    //SDK引擎库的路径
    private String libPath;
    //SDK APP_ID
    private String appId;
    //SDK SDK_KEY
    private String sdkKey;
    //SDK 激活码
    private String activeKey;
    //引擎配置类
    private EngineConfiguration engineConfiguration;
    

    2)create()方法

    @Override
    public FaceEngine create() {
        FaceEngine faceEngine = new FaceEngine(libPath);
        int activeCode = faceEngine.activeOnline(appId, sdkKey);
        if (activeCode != ErrorInfo.MOK.getValue() && activeCode != ErrorInfo.MERR_ASF_ALREADY_ACTIVATED.getValue()) {
            log.error("引擎激活失败" + activeCode);
            throw new BusinessException(ErrorCodeEnum.FAIL, "引擎激活失败" + activeCode);
        }
        int initCode = faceEngine.init(engineConfiguration);
        if (initCode != ErrorInfo.MOK.getValue()) {
            log.error("引擎初始化失败" + initCode);
            throw new BusinessException(ErrorCodeEnum.FAIL, "引擎初始化失败" + initCode);
        }
        return faceEngine;
    }
    
    1. 参数说明:无

    2. 返回结果:FaceEngine人脸识别引擎

    3. 代码流程解读:

      此方法,通过libPath(SDK引擎库的路径)实例化FaceEngine对象,再根据APP_IDSDK_KEY调用activeOnline()方法激活引擎 (联网状态下)

      成功激活引擎后,根据EngineConfiguration引擎配置类 调用init()方法初始化引擎 。

    3)wrap()方法

    public PooledObject<FaceEngine> wrap(FaceEngine faceEngine) {
        return new DefaultPooledObject<>(faceEngine);
    }
    
    1. 参数说明:FaceEngine人脸识别引擎

    2. 返回结果:PooledObject包装类

    3. 代码流程解读:

      此方法,通过PooledObject包装器对象 将faceEngine进行包装,便于维护引擎的状态。

    4)destroyObject()方法

    public void destroyObject(PooledObject<FaceEngine> p) throws Exception {
        FaceEngine faceEngine = p.getObject();
        int result = faceEngine.unInit();
        super.destroyObject(p);
    }
    
    1. 参数说明:PooledObject包装类

    2. 返回结果:无

    3. 代码流程解读:

      此方法,从PooledObject包装器对象中获取faceEngine引擎,随后卸载引擎。

    4. FaceEngineServiceImpl类 源码说明

    1)成员变量说明

    @Value("${config.arcface-sdk.sdk-lib-path}")
    public String sdkLibPath;
    
    @Value("${config.arcface-sdk.app-id}")
    public String appId;
    
    @Value("${config.arcface-sdk.sdk-key}")
    public String sdkKey;
    
    @Value("${config.arcface-sdk.detect-pool-size}")
    public Integer detectPooSize;
    
    @Value("${config.arcface-sdk.compare-pool-size}")
    public Integer comparePooSize;
    
    private ExecutorService compareExecutorService;
    
    //通用人脸识别引擎池
    private GenericObjectPool<FaceEngine> faceEngineGeneralPool;
    
    //人脸比对引擎池
    private GenericObjectPool<FaceEngine> faceEngineComparePool;
    

    ​ 此类的成员变量可通过@Value()注解获取配置文件中的相应配置。

    2)init()方法

    @PostConstruct
    public void init() {
        GenericObjectPoolConfig detectPoolConfig = new GenericObjectPoolConfig();
        detectPoolConfig.setMaxIdle(detectPooSize);
        detectPoolConfig.setMaxTotal(detectPooSize);
        detectPoolConfig.setMinIdle(detectPooSize);
        detectPoolConfig.setLifo(false);
        EngineConfiguration detectCfg = new EngineConfiguration();
        FunctionConfiguration detectFunctionCfg = new FunctionConfiguration();
        //开启人脸检测功能
        detectFunctionCfg.setSupportFaceDetect(true);
        //开启人脸识别功能
        detectFunctionCfg.setSupportFaceRecognition(true);
        //开启年龄检测功能
        detectFunctionCfg.setSupportAge(true);
        //开启性别检测功能
        detectFunctionCfg.setSupportGender(true);
        //开启活体检测功能
        detectFunctionCfg.setSupportLiveness(true);
        detectCfg.setFunctionConfiguration(detectFunctionCfg);
        //图片检测模式,如果是连续帧的视频流图片,那么改成VIDEO模式
        detectCfg.setDetectMode(DetectMode.ASF_DETECT_MODE_IMAGE);
        //人脸旋转角度
        detectCfg.setDetectFaceOrientPriority(DetectOrient.ASF_OP_0_ONLY);
        //底层库算法对象池
        faceEngineGeneralPool = new GenericObjectPool(new FaceEngineFactory(sdkLibPath, appId, sdkKey, null, detectCfg), detectPoolConfig);
    
        //初始化特征比较线程池
        GenericObjectPoolConfig comparePoolConfig = new GenericObjectPoolConfig();
        comparePoolConfig.setMaxIdle(comparePooSize);
        comparePoolConfig.setMaxTotal(comparePooSize);
        comparePoolConfig.setMinIdle(comparePooSize);
        comparePoolConfig.setLifo(false);
        EngineConfiguration compareCfg = new EngineConfiguration();
        FunctionConfiguration compareFunctionCfg = new FunctionConfiguration();
        //开启人脸识别功能
        compareFunctionCfg.setSupportFaceRecognition(true);
        compareCfg.setFunctionConfiguration(compareFunctionCfg);
        //图片检测模式,如果是连续帧的视频流图片,那么改成VIDEO模式
        compareCfg.setDetectMode(DetectMode.ASF_DETECT_MODE_IMAGE);
        //人脸旋转角度
        compareCfg.setDetectFaceOrientPriority(DetectOrient.ASF_OP_0_ONLY);
        //底层库算法对象池
        faceEngineComparePool = new GenericObjectPool(new FaceEngineFactory(sdkLibPath, appId, sdkKey, null, compareCfg), comparePoolConfig);
        compareExecutorService = Executors.newFixedThreadPool(comparePooSize);
    }
    
    1. 参数说明:无

    2. 返回结果:无

    3. 代码流程解读:

      @PostConstruct注解:Spring在实例化该Bean之后 立刻去执行此方法。在此方法中,首先去实例化通用人脸识别引擎池配置对象并设置其对应属性,之后实例化EngineConfiguration(设置图像检测模式、人脸旋转角度)和FunctionConfiguration(用于功能配置,开启引擎相应功能,被EngineConfiguration所依赖),最后调FaceEngineFactory的构造方法去初始化引擎并获取对象池。人脸比对引擎池 同理。

    3)detectFaces()方法 人脸检测

    @Override
    public List<FaceInfo> detectFaces(ImageInfo imageInfo) {
    
        FaceEngine faceEngine = null;
        try {
            faceEngine = faceEngineGeneralPool.borrowObject();
            if (faceEngine == null) {
                throw new BusinessException(ErrorCodeEnum.FAIL, "获取引擎失败");
            }
    
            //人脸检测得到人脸列表
            List<FaceInfo> faceInfoList = new ArrayList<FaceInfo>();
            //人脸检测
            int errorCode = faceEngine.detectFaces(imageInfo.getImageData(), imageInfo.getWidth(), imageInfo.getHeight(), imageInfo.getImageFormat(), faceInfoList);
            if (errorCode == 0) {
                return faceInfoList;
            } else {
                log.error("人脸检测失败,errorCode:" + errorCode);
            }
    
        } catch (Exception e) {
            log.error("", e);
        } finally {
            if (faceEngine != null) {
                //释放引擎对象
                faceEngineGeneralPool.returnObject(faceEngine);
            }
        }
    
        return null;
    
    }
    
    1. 参数说明:ImageInfo图像信息

    2. 返回结果:List<FaceInfo> 人脸信息列表

    3. 代码流程解读:

      此方法,根据传入的ImageInfo图像信息,通过faceEngine引擎调用detectFaces()方法检测人脸信息(所需参数: 图像数据,图像宽度(4的倍数),图片高度,图像的颜色格式,存放检测到的人脸信息List),随后回收引擎对象。

      注:detectFaces该功能依赖初始化的模式选择,初始化中detectFaceOrientPrioritydetectFaceScaleValdetectFaceMaxNum参数的设置,对能否检测到人脸以及检测到几张人脸都有决定性的作用。

    4)extractFaceFeature()方法 人脸特征值提取

    @Override
    public byte[] extractFaceFeature(ImageInfo imageInfo, FaceInfo faceInfo) {
    
        FaceEngine faceEngine = null;
        try {
            faceEngine = faceEngineGeneralPool.borrowObject();
            if (faceEngine == null) {
                throw new BusinessException(ErrorCodeEnum.FAIL, "获取引擎失败");
            }
    
            FaceFeature faceFeature = new FaceFeature();
            //提取人脸特征
            int errorCode = faceEngine.extractFaceFeature(imageInfo.getImageData(), imageInfo.getWidth(), imageInfo.getHeight(), imageInfo.getImageFormat(), faceInfo, faceFeature);
            if (errorCode == 0) {
                return faceFeature.getFeatureData();
            } else {
                log.error("特征提取失败,errorCode:" + errorCode);
            }
    
        } catch (Exception e) {
            log.error("", e);
        } finally {
            if (faceEngine != null) {
                //释放引擎对象
                faceEngineGeneralPool.returnObject(faceEngine);
            }
        }
    
        return null;
    
    }
    
    1. 参数说明:ImageInfo图像信息,FaceInfo人脸信息

    2. 返回结果:人脸特征值 字节数组

    3. 代码流程解读:

      此方法,根据传入的ImageInfo图像信息数据和FaceInfo人脸信息 通过faceEngine引擎调用extractFaceFeature()方法获取人脸特征数据(所需参数:图像数据,图像宽度(4的倍数),图像高度,图像的颜色格式,人脸信息,存放提取到的人脸特征信息),随后回收引擎对象。

      注:extractFaceFeature()方法依赖detectFaces成功检测到人脸,将检测到的人脸,取单张人脸信息和使用的图像信息 传入该接口进行特征提取。

    5)compareFace()方法 人脸相似度比对

    @Override
    public Float compareFace(ImageInfo imageInfo1, ImageInfo imageInfo2) {
    
        List<FaceInfo> faceInfoList1 = detectFaces(imageInfo1);
        List<FaceInfo> faceInfoList2 = detectFaces(imageInfo2);
    
        if (CollectionUtil.isEmpty(faceInfoList1) || CollectionUtil.isEmpty(faceInfoList2)) {
            throw new BusinessException(ErrorCodeEnum.FAIL,"未检测到人脸");
        }
    
        byte[] feature1 = extractFaceFeature(imageInfo1, faceInfoList1.get(0));
        byte[] feature2 = extractFaceFeature(imageInfo2, faceInfoList2.get(0));
    
        FaceEngine faceEngine = null;
        try {
            faceEngine = faceEngineGeneralPool.borrowObject();
            if (faceEngine == null) {
                throw new BusinessException(ErrorCodeEnum.FAIL, "获取引擎失败");
            }
    
            FaceFeature faceFeature1 = new FaceFeature();
            faceFeature1.setFeatureData(feature1);
            FaceFeature faceFeature2 = new FaceFeature();
            faceFeature2.setFeatureData(feature2);
            //提取人脸特征
            FaceSimilar faceSimilar = new FaceSimilar();
            int errorCode = faceEngine.compareFaceFeature(faceFeature1, faceFeature2, faceSimilar);
            if (errorCode == 0) {
                return faceSimilar.getScore();
            } else {
                log.error("特征提取失败,errorCode:" + errorCode);
            }
    
        } catch (Exception e) {
            log.error("", e);
        } finally {
            if (faceEngine != null) {
                //释放引擎对象
                faceEngineGeneralPool.returnObject(faceEngine);
            }
        }
    
        return null;
    
    }
    
    1. 参数说明:需要比对的两个 ImageInfo图像信息

    2. 返回结果:人脸比对相似值

    3. 代码流程解读:

      此方法,根据传入的两个ImageInfo图像信息分别调用detectFaces()方法获取各自人脸信息,成功检测到人脸信息后,再调用extractFaceFeature()方法提取各自人脸特征值,成功获取到人脸特征值后,根据两个特征值再通过faceEngine引擎调用compareFaceFeature()方法进行比对(所需参数:人脸特征值1,人脸特征值2,比对模型,存放比对相似值结果), 最后获取人脸相似值返回 并回收引擎对象。

    6)CompareFaceTask

    FaceEngineServiceImpl的一个成员内部类,其实现Callable接口,用于完成线程任务

    private class CompareFaceTask implements Callable<List<UserCompareInfo>> {
        //存储的人脸信息列表
        private List<UserRamCache.UserInfo> userInfoList;
        //目标特征值
        private FaceFeature targetFaceFeature;
        //相似度预期值
        private float passRate;
    
        public CompareFaceTask(List<UserRamCache.UserInfo> userInfoList, FaceFeature targetFaceFeature, float passRate) {
            this.userInfoList = userInfoList;
            this.targetFaceFeature = targetFaceFeature;
            this.passRate = passRate;
        }
    
        @Override
        public List<UserCompareInfo> call() throws Exception {
            FaceEngine faceEngine = null;
            List<UserCompareInfo> resultUserInfoList = Lists.newLinkedList();//识别到的人脸列表
            try {
                faceEngine = faceEngineComparePool.borrowObject();
                for (UserRamCache.UserInfo userInfo : userInfoList) {
                    FaceFeature sourceFaceFeature = new FaceFeature();
                    sourceFaceFeature.setFeatureData(userInfo.getFaceFeature());
                    FaceSimilar faceSimilar = new FaceSimilar();
                    faceEngine.compareFaceFeature(targetFaceFeature, sourceFaceFeature, faceSimilar);
                    if (faceSimilar.getScore() > passRate) {//相似值大于配置预期,加入到识别到人脸的列表
                        UserCompareInfo info = new UserCompareInfo();
                        info.setName(userInfo.getName());
                        info.setFaceId(userInfo.getFaceId());
                        info.setSimilar(faceSimilar.getScore());
                        resultUserInfoList.add(info);
                    }
                }
            } catch (Exception e) {
                logger.error("", e);
            } finally {
                if (faceEngine != null) {
                    faceEngineComparePool.returnObject(faceEngine);
                }
            }
            return resultUserInfoList;
        }
    }
    
    1. 参数说明:无

    2. 返回结果:List<UserCompareInfo>人脸识别实体类列表

    3. 代码流程解读:

      call()方法中遍历userInfoList中每个UserInfo获取特征值,并结合目标特征值 通过faceEngine引擎调用compareFaceFeature()方法获取相似度大小 ,将获取到的相似度大小和预期相似度进行比较,若大于配置的预期值,则加入到识别到人脸的列表,最后回收引擎对象。

    7)faceRecognition()方法 人脸识别

    @Override
    public List<UserCompareInfo> faceRecognition(byte[] faceFeature, List<UserRamCache.UserInfo> userInfoList, float passRate) {
        List<UserCompareInfo> resultUserInfoList = Lists.newLinkedList();//识别到的人脸列表
    
        FaceFeature targetFaceFeature = new FaceFeature();
        targetFaceFeature.setFeatureData(faceFeature);
    
        List<List<UserRamCache.UserInfo>> faceUserInfoPartList = Lists.partition(userInfoList, 1000);//分成1000一组,多线程处理
        CompletionService<List<UserCompareInfo>> completionService = new ExecutorCompletionService(compareExecutorService);
        for (List<UserRamCache.UserInfo> part : faceUserInfoPartList) {
            completionService.submit(new CompareFaceTask(part, targetFaceFeature, passRate));
        }
        for (int i = 0; i < faceUserInfoPartList.size(); i++) {
            List<UserCompareInfo> faceUserInfoList = null;
            try {
                faceUserInfoList = completionService.take().get();
            } catch (InterruptedException | ExecutionException e) {
            }
            if (CollectionUtil.isNotEmpty(userInfoList)) {
                resultUserInfoList.addAll(faceUserInfoList);
            }
        }
    
        resultUserInfoList.sort((h1, h2) -> h2.getSimilar().compareTo(h1.getSimilar()));//从大到小排序
    
        return resultUserInfoList;
    }
    
    1. 参数说明:人脸特征值 字节数组,List<UserRamCache.UserInfo>存储的人脸信息列表,相似度预期值

    2. 返回结果:List<UserCompareInfo>人脸识别实体类列表

    3. 代码流程解读:

      此方法,根据传入的人脸特征值以及UserRamCache.UserInfo列表,先将UserRamCache.UserInfo分为每1000一组,再通过多线程处理,即上述CompareFaceTask类,处理完之后再将结果合并,按从大到小排序后返回。

    8)process()方法 人脸属性检测

    @Override
    public List<ProcessInfo> process(ImageInfo imageInfo, List<FaceInfo> faceInfoList) {
        FaceEngine faceEngine = null;
        try {
            //获取引擎对象
            faceEngine = faceEngineGeneralPool.borrowObject();
            if (faceEngine == null) {
                throw new BusinessException(ErrorCodeEnum.FAIL, "获取引擎失败");
            }
    
            int errorCode = faceEngine.process(imageInfo.getImageData(), imageInfo.getWidth(), imageInfo.getHeight(), imageInfo.getImageFormat(), faceInfoList, FunctionConfiguration.builder().supportAge(true).supportGender(true).supportLiveness(true).build());
            if (errorCode == 0) {
                List<ProcessInfo> processInfoList = Lists.newLinkedList();
    
                //性别列表
                List<GenderInfo> genderInfoList = new ArrayList<GenderInfo>();
                faceEngine.getGender(genderInfoList);
    
                //年龄列表
                List<AgeInfo> ageInfoList = new ArrayList<AgeInfo>();
                faceEngine.getAge(ageInfoList);
                //活体结果列表
                List<LivenessInfo> livenessInfoList = new ArrayList<LivenessInfo>();
                faceEngine.getLiveness(livenessInfoList);
    
    
                for (int i = 0; i < genderInfoList.size(); i++) {
                    ProcessInfo processInfo = new ProcessInfo();
                    processInfo.setGender(genderInfoList.get(i).getGender());
                    processInfo.setAge(ageInfoList.get(i).getAge());
                    processInfo.setLiveness(livenessInfoList.get(i).getLiveness());
                    processInfoList.add(processInfo);
                }
                return processInfoList;
    
            }
        } catch (Exception e) {
            logger.error("", e);
        } finally {
            if (faceEngine != null) {
                //释放引擎对象
                faceEngineGeneralPool.returnObject(faceEngine);
            }
        }
    
        return null;
    
    }
    
    1. 参数说明:ImageInfo图像信息,List<FaceInfo>人脸信息列表

    2. 返回结果:List<ProcessInfo>人脸检测实体类列表

    3. 代码流程解读:

      此方法,根据传入的ImageInfo图像信息以及检测到的FaceInfo人脸信息列表,通过faceEngine引擎调用process()方法(所需参数:图像数据,图片宽度(4的倍数),图像高度,图像的颜色空间格式,人脸信息列表,需检测的属性),之后可从faceEngine对象获取 性别,年龄,是否活体等结果的列表,并将一系列列表结果遍历 设置于ProcessInfo对象返回,随后回收引擎对象。

      注:process()支持检测AGEGENDERFACE3DANGLELIVENESS四种属性,若想检测这些属性,须在初始化引擎接口中对想要检测的属性进行设置。

    5. FaceController类 源码说明

    1)initFace()方法:初始化注册人脸

    @PostConstruct
    public void initFace() throws FileNotFoundException {
        Map<String, String> fileMap = Maps.newHashMap();
        fileMap.put("zhao1", "赵丽颖");
        fileMap.put("yang1", "杨紫");
    
        for (String f : fileMap.keySet()) {
            ClassPathResource resource = new ClassPathResource("static/images/" + f +  ".jpg");
            InputStream inputStream = null;
            try {
                inputStream = resource.getInputStream();
            } catch (IOException e) {
            }
            ImageInfo rgbData = ImageFactory.getRGBData(inputStream);
            List<FaceInfo> faceInfoList = faceEngineService.detectFaces(rgbData);
            if (CollectionUtil.isNotEmpty(faceInfoList)) {
                byte[] feature = faceEngineService.extractFaceFeature(rgbData, faceInfoList.get(0));
                UserRamCache.UserInfo userInfo = new UserCompareInfo();
                userInfo.setFaceId(f);
                userInfo.setName(fileMap.get(f));
                userInfo.setFaceFeature(feature);
                UserRamCache.addUser(userInfo);
            }
        }
    
        log.info("http://127.0.0.1:" + port + "/");
    
    }
    
    1. 参数说明:无

    2. 返回结果:无

    3. 代码流程解读:

      @PostConstruct注解表示Spring在实例化该Bean之后 立刻去执行此方法。首先去加载static/images/下的图片资源将其解析为ImageInfo类型的RGB图像信息数据,之后依次调用FaceEngineService类的detectFaces()extractFaceFeature()方法提取人脸特征值,最后将人脸相关数据设置于UserRamCache.UserInfo对象中(此Demo仅将数据暂存于内存中,用户可根据需要,自行创建数据库相关表并持久化于磁盘中)。

    2)faceAdd()方法 添加人脸

    @RequestMapping(value = "/faceAdd", method = RequestMethod.POST)
    @ResponseBody
    public Response faceAdd(String file, String faceId, String name) {
        return null;
    }
    
    1. 参数说明:浏览器上传的图片信息,人脸Id,人脸名

    2. 返回结果:Json格式

    3. 代码流程解读:

      此方法,可用于添加更多人脸信息,根据用户需要自行完善。

    3)faceRecognition()方法 人脸识别

    @RequestMapping(value = "/faceRecognition", method = RequestMethod.POST)
    @ResponseBody
    public Response<List<FaceRecognitionResDTO>> faceRecognition(String image) {
    
        List<FaceRecognitionResDTO> faceRecognitionResDTOList = Lists.newLinkedList();
        byte[] bytes = Base64Util.base64ToBytes(image);
        ImageInfo rgbData = ImageFactory.getRGBData(bytes);
        List<FaceInfo> faceInfoList = faceEngineService.detectFaces(rgbData);
        if (CollectionUtil.isNotEmpty(faceInfoList)) {
            for (FaceInfo faceInfo : faceInfoList) {
                FaceRecognitionResDTO faceRecognitionResDTO = new FaceRecognitionResDTO();
                faceRecognitionResDTO.setRect(faceInfo.getRect());
                byte[] feature = faceEngineService.extractFaceFeature(rgbData, faceInfo);
                if (feature != null) {
                    List<UserCompareInfo> userCompareInfos = faceEngineService.faceRecognition(feature, UserRamCache.getUserList(), 0.8f);
                    if (CollectionUtil.isNotEmpty(userCompareInfos)) {
                        faceRecognitionResDTO.setName(userCompareInfos.get(0).getName());
                        faceRecognitionResDTO.setSimilar(userCompareInfos.get(0).getSimilar());
                    }
                }
                faceRecognitionResDTOList.add(faceRecognitionResDTO);
            }
        }
        return Response.newSuccessResponse(faceRecognitionResDTOList);
    }
    
    1. 参数说明:浏览器上传的图片信息

    2. 返回结果:Json格式 List<FaceRecognitionResDTO>人脸识别DTO列表

    3. 代码流程解读:

      此方法,先将请求上传的(base64编码后)的图片解析为ImageInfo类型的RGB图像信息数据,再依次调用FaceEngineService类的detectFaces()extractFaceFeature()faceRecognition()方法 与先前存于内存中的人脸信息进行比对,获取相似度最大的人脸信息,并将结果设置于FaceRecognitionResDTO后返回。

    4)detectFaces() 方法 人脸检测

    @RequestMapping(value = "/detectFaces", method = RequestMethod.POST)
    @ResponseBody
    public Response<List<FaceDetectResDTO>> detectFaces(String image) {
    
        byte[] bytes = Base64Util.base64ToBytes(image);
        ImageInfo rgbData = ImageFactory.getRGBData(bytes);
        List<FaceDetectResDTO> faceDetectResDTOS = Lists.newLinkedList();
        List<FaceInfo> faceInfoList = faceEngineService.detectFaces(rgbData);
        if (CollectionUtil.isNotEmpty(faceInfoList)) {
            List<ProcessInfo> process = faceEngineService.process(rgbData, faceInfoList);
    
            for (int i = 0; i < faceInfoList.size(); i++) {
                FaceDetectResDTO faceDetectResDTO = new FaceDetectResDTO();
                FaceInfo faceInfo = faceInfoList.get(i);
                faceDetectResDTO.setRect(faceInfo.getRect());
                faceDetectResDTO.setOrient(faceInfo.getOrient());
                faceDetectResDTO.setFaceId(faceInfo.getFaceId());
                if (CollectionUtil.isNotEmpty(process)) {
                    ProcessInfo processInfo = process.get(i);
                    faceDetectResDTO.setAge(processInfo.getAge());
                    faceDetectResDTO.setGender(processInfo.getGender());
                    faceDetectResDTO.setLiveness(processInfo.getLiveness());
    
                }
                faceDetectResDTOS.add(faceDetectResDTO);
    
            }
        }
    
        return Response.newSuccessResponse(faceDetectResDTOS);
    }
    
    1. 参数说明:浏览器上传的图片信息

    2. 返回结果:Json格式 List<FaceDetectResDTO>人脸检测DTO列表

    3. 代码流程解读:

      此方法,先将请求上传的(base64编码后)的图片解析为ImageInfo类型的RGB图像信息数据,再依次调用FaceEngineService类的detectFaces() process()方法获取人脸检测数据(年龄,性别,是否活体),并将结果设置于FaceDetectResDTO后返回。

    5)compareFaces()方法 人脸比对

    @RequestMapping(value = "/compareFaces", method = RequestMethod.POST)
    @ResponseBody
    public Response<Float> compareFaces(String image1, String image2) {
    
        byte[] bytes1 = Base64Util.base64ToBytes(image1);
        byte[] bytes2 = Base64Util.base64ToBytes(image2);
        ImageInfo rgbData1 = ImageFactory.getRGBData(bytes1);
        ImageInfo rgbData2 = ImageFactory.getRGBData(bytes2);
    
        Float similar = faceEngineService.compareFace(rgbData1, rgbData2);
    
        return Response.newSuccessResponse(similar);
    }
    
    1. 参数说明:两张浏览器上传的图片信息

    2. 返回结果:人脸比对相似值

    3. 代码流程解读:

      此方法,先将请求上传的(base64编码后)的图片解析为ImageInfo类型的RGB图像信息数据,之后通过FaceEngineService类的compareFace()方法进行人脸比对,获取人脸相似值并返回。

    七、源码下载

    ​ 若有想一起学习虹软SDK,感受人脸识别奥秘的同学,可通过点击此链接获取Demo源码。
    相关产品大家可以到虹软视觉开放平台进一步了解

  • 相关阅读:
    django之contenttype
    __new__ 与 __init__的区别
    __getattr__ 与 __getattribute__的区别
    wsgi 简介
    restframework
    ASPxGridView改变列颜色
    C#导出带有格式的Excel(列宽,合并单元格,显示边框线,加背景颜色等)
    c#创建、保存excel正常执行
    C# 导出 Excel 和相关打印设置
    C#命名空间“Microsoft.Office”中不存在类型或命名空间名称的终极解决方法
  • 原文地址:https://www.cnblogs.com/ccLqqy/p/13606987.html
Copyright © 2011-2022 走看看