宏软--人脸比对 --- java代码---AfcFace2.0

首先去虹软官网注册获取对应的

官网下载对应的 jar ,导入到对应的项目的  libs中

pom.xml

<dependency>
    <groupId>arcsoft</groupId>
    <artifactId>arcsoft</artifactId>
    <version>2.0</version>
    <scope>system</scope>
    <systemPath>${basedir}/lib/arcsoft-sdk-face-2.0.0.0.jar</systemPath>
</dependency>

<build>
    <plugins>
        <plugin>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-maven-plugin</artifactId>
        </plugin>
    </plugins>
    <resources>
        <resource>
            <directory>src/main/resources</directory>
        </resource>
        <resource>
            <directory>lib</directory>
            <targetPath>BOOT-INF/lib/</targetPath>
            <includes>
                <include>**/*.jar</include>
            </includes>
        </resource>
    </resources>
</build>

String APPID="*********************************************";

String SDKKEY="*********************************************";

/**
 *  添加测试人脸数据
 */
@RequestMapping("/addTestFace")
    public  void addTestFace(){
        try {
            FaceEngine faceEngine = new FaceEngine();
            //激活引擎
            faceEngine.active(APPID, SDKKEY);

            EngineConfiguration engineConfiguration = EngineConfiguration.builder().functionConfiguration(
                    FunctionConfiguration.builder()
                            .supportAge(true)
                            .supportFace3dAngle(true)
                            .supportFaceDetect(true)
                            .supportFaceRecognition(true)
                            .supportGender(true)
                            .build()).build();
            //初始化引擎
            faceEngine.init(engineConfiguration);

            List<String> lists = new ArrayList<>();

            ImageInfo imageInfo = getRGBData(new File("C:\\Users\\Administrator\\Desktop\\x-1.jpg"));
            //人脸检测
            List<FaceInfo> faceInfoList = new ArrayList<FaceInfo>();
            faceEngine.detectFaces(imageInfo.getRgbData(), imageInfo.getWidth(), imageInfo.getHeight(), ImageFormat.CP_PAF_BGR24, faceInfoList);

            //提取人脸特征
            FaceFeature faceFeature = new FaceFeature();
            faceEngine.extractFaceFeature(imageInfo.getRgbData(), imageInfo.getWidth(), imageInfo.getHeight(), ImageFormat.CP_PAF_BGR24, faceInfoList.get(0), faceFeature);

            lists.add(UtilHelper.byte2Base64StringFun(faceFeature.getFeatureData()));

            List<String> list = new ArrayList<>();
            for (int i = 1; i < 2000; i++){
                if(i==12){
                    continue;
                }
                redisUtil.set(i + "", lists, 60 *10);
                list.add(i + "");
            }
            redisUtil.set("12" + "-" + "A" + "-" + "4", list);
        }catch (Exception e){
            e.printStackTrace();
        }
    }


    /**
     * 后台添加人脸数据
     * @param imagePath  人脸图片路径 多个人脸用   ,  分割
     * @param userId     用户ID
     * @param districtId 项目ID
     * @param floorId    楼栋ID
     * @param unitId     单元ID
     * @return           返回是否添加成功结果
     */
    @RequestMapping("/addFace")
    public BaseResult addFace(String imagePath, String userId, String districtId, String floorId, String unitId){

        System.out.println("---------------------开始添加人脸数据---------------------");

        if(imagePath == null || districtId == null){
            return new BaseResult("0", "图片路径或者项目ID不能为空",null);
        }

        if((floorId == null && unitId != null) || (floorId != null && unitId == null)){
            return new BaseResult("0", "栋和单元应同时存在或者不存在",null);
        }

        try {
            FaceEngine faceEngine = new FaceEngine();
            //激活引擎
            faceEngine.active(APPID, SDKKEY);

            EngineConfiguration engineConfiguration = EngineConfiguration.builder().functionConfiguration(
                    FunctionConfiguration.builder()
                            .supportAge(true)
                            .supportFace3dAngle(true)
                            .supportFaceDetect(true)
                            .supportFaceRecognition(true)
                            .supportGender(true)
                            .build()).build();
            //初始化引擎
            faceEngine.init(engineConfiguration);

//            System.out.println("图片路径:" + imagePath);
            //获取传递过来的人脸图片信息，并保存到redis中去
            String[] imgPath = imagePath.split(",");

            List<String> lists = new ArrayList<>();
            for (String img : imgPath) {

                String imageUrl = ImageUtils.image(img);
                File file = new File(imageUrl);
                ImageInfo imageInfo = getRGBData(file);
                //人脸检测
                List<FaceInfo> faceInfoList = new ArrayList<FaceInfo>();
                faceEngine.detectFaces(imageInfo.getRgbData(), imageInfo.getWidth(), imageInfo.getHeight(), ImageFormat.CP_PAF_BGR24, faceInfoList);

                //提取人脸特征
                FaceFeature faceFeature = new FaceFeature();
                faceEngine.extractFaceFeature(imageInfo.getRgbData(), imageInfo.getWidth(), imageInfo.getHeight(), ImageFormat.CP_PAF_BGR24, faceInfoList.get(0), faceFeature);

                lists.add(UtilHelper.byte2Base64StringFun(faceFeature.getFeatureData()));

                file.delete();
            }
            redisUtil.set(userId, lists);

            if(floorId == null || unitId == null){
                //非小区模式的情况下，存储在
                List<String> districtUserList = (List<String>)redisUtil.get(districtId);
                if(districtUserList == null){
                    districtUserList = new ArrayList<>();
                }
                if(!districtUserList.contains(userId)){
                    districtUserList.add(userId);
                    redisUtil.set(districtId, districtUserList);
                }
            }else{
                //小区模式

                //小区
                List<String> districtUserList = (List<String>)redisUtil.get(districtId);
                if(districtUserList == null){
                    districtUserList = new ArrayList<>();
                }
                if(!districtUserList.contains(floorId)){
                    districtUserList.add(floorId);
                }
                redisUtil.set(districtId, districtUserList);

                //小区-栋
                List<String> districtFloor = (List<String>)redisUtil.get(districtId + "-" + floorId);
                if(districtFloor == null){
                    districtFloor = new ArrayList<>();
                }
                if(!districtFloor.contains(unitId)){
                    districtFloor.add(unitId);
                }
                redisUtil.set(districtId + "-" + floorId, districtFloor);

                //小区-栋-单元
                List<String> districtFloorUnit = (List<String>)redisUtil.get(districtId + "-" + floorId + "-" + unitId);
                if(districtFloorUnit == null){
                    districtFloorUnit = new ArrayList<>();
                }
                if(!districtFloorUnit.contains(userId)){
                    districtFloorUnit.add(userId);
                }
                redisUtil.set(districtId + "-" + floorId + "-" + unitId, districtFloorUnit);
            }

            System.out.println("成功");

            return new BaseResult("1", "成功",lists);
        }catch (Exception e){
            e.printStackTrace();
            return new BaseResult("0", "异常",null);
        }

    }

/**
 *  districtId  小区ID
 *  imagePath  图片路径
 *  raft 筏值
 * @param request
 * @return
 * @throws Exception
 */
@RequestMapping("/check")
public BaseResult check(HttpServletRequest request){
    try {
        long startTime = System.currentTimeMillis();
        String districtId = request.getParameter("districtId");
        String imagePath = request.getParameter("imagePath");
        String raft = request.getParameter("raft");

        Map<String, Float> list = new HashMap<>();

        FaceEngine faceEngine = new FaceEngine();
        //激活引擎
        faceEngine.active(APPID, SDKKEY);

        EngineConfiguration engineConfiguration = EngineConfiguration.builder().functionConfiguration(
                FunctionConfiguration.builder()
                        .supportAge(true)
                        .supportFace3dAngle(true)
                        .supportFaceDetect(true)
                        .supportFaceRecognition(true)
                        .supportGender(true)
                        .build()).build();
        //初始化引擎
        faceEngine.init(engineConfiguration);

        String imageUrl = ImageUtils.image(imagePath);
        File file = new File(imageUrl);
        ImageInfo imageInfo = getRGBData(file);
        file.delete();

        //目标人脸检测
        List<FaceInfo> faceInfoList = new ArrayList<FaceInfo>();
        faceEngine.detectFaces(imageInfo.getRgbData(), imageInfo.getWidth(), imageInfo.getHeight(), ImageFormat.CP_PAF_BGR24, faceInfoList);

        List<String> floorList = (List<String>)redisUtil.get(districtId);
        if(floorList == null){
            return new BaseResult("0", "小区信息不存在, 比对识别",null);
        }

        for (FaceInfo faceInfo: faceInfoList) {
            //目标提取人脸特征
            FaceFeature faceFeature = new FaceFeature();
            faceEngine.extractFaceFeature(imageInfo.getRgbData(), imageInfo.getWidth(), imageInfo.getHeight(), ImageFormat.CP_PAF_BGR24, faceInfo, faceFeature);

            //人脸对比
            FaceFeature targetFaceFeature = new FaceFeature();
            targetFaceFeature.setFeatureData(faceFeature.getFeatureData());

            for (String floorId: floorList) {
                List<String> unitList = (List<String>)redisUtil.get(districtId + "-" + floorId);
                if(unitList != null){
                    for (String unitId : unitList) {
                        List<String> userList = (List<String>)redisUtil.get(districtId + "-" + floorId + "-" + unitId);

                        if(userList != null){
                            for (String userId : userList) {

                                List<String> faceFeatures = (List<String>)redisUtil.get(userId);
                                if(faceFeatures != null){
                                    for (String faceBase: faceFeatures) {

                                        //人脸对比
                                        FaceFeature sourceFaceFeature = new FaceFeature();
                                        sourceFaceFeature.setFeatureData(UtilHelper.base64String2ByteFun(faceBase));

                                        FaceSimilar faceSimilar = new FaceSimilar();
                                        faceEngine.compareFaceFeature(targetFaceFeature, sourceFaceFeature, faceSimilar);

                                        if(faceSimilar.getScore() * 100 > Integer.valueOf(raft)){

                                            long endTime = System.currentTimeMillis();    //获取结束时间
                                            System.out.println("程序运行时间：" + (endTime - startTime) + "ms");    //输出程序运行时间
                                            list.put(userId, faceSimilar.getScore() * 100);
                                        }
                                    }
                                }
                            }
                        }
                    }
                }
            }
        }

        long endTime = System.currentTimeMillis();    //获取结束时间
        System.out.println("程序运行时间：" + (endTime - startTime) + "ms");    //输出程序运行时间

        return new BaseResult("1", "识别成功",list);
    }catch (Exception e){
        e.printStackTrace();
        return new BaseResult("0", "匹配失败",null);
    }
}

public ImageInfo getRGBData(File file) {
    if (file == null)
        return null;
    ImageInfo imageInfo;
    try {
        //将图片文件加载到内存缓冲区
        BufferedImage image = ImageIO.read(file);
        imageInfo = bufferedImage2ImageInfo(image);
    } catch (IOException e) {
        e.printStackTrace();
        return null;
    }
    return imageInfo;
}

private ImageInfo bufferedImage2ImageInfo(BufferedImage image) {
    ImageInfo imageInfo = new ImageInfo();
    int width = image.getWidth();
    int height = image.getHeight();
    // 使图片居中
    width = width & (~3);
    height = height & (~3);
    imageInfo.width = width;
    imageInfo.height = height;
    //根据原图片信息新建一个图片缓冲区
    BufferedImage resultImage = new BufferedImage(width, height, image.getType());
    //得到原图的rgb像素矩阵
    int[] rgb = image.getRGB(0, 0, width, height, null, 0, width);
    //将像素矩阵 绘制到新的图片缓冲区中
    resultImage.setRGB(0, 0, width, height, rgb, 0, width);
    //进行数据格式化为可用数据
    BufferedImage dstImage = new BufferedImage(image.getWidth(), image.getHeight(), BufferedImage.TYPE_3BYTE_BGR);
    if (resultImage.getType() != BufferedImage.TYPE_3BYTE_BGR) {
        ColorSpace cs = ColorSpace.getInstance(ColorSpace.CS_LINEAR_RGB);
        ColorConvertOp colorConvertOp = new ColorConvertOp(cs, dstImage.createGraphics().getRenderingHints());
        colorConvertOp.filter(resultImage, dstImage);
    } else {
        dstImage = resultImage;
    }

    //获取rgb数据
    imageInfo.rgbData = ((DataBufferByte) (dstImage.getRaster().getDataBuffer())).getData();
    return imageInfo;
}


class ImageInfo {
    public byte[] rgbData;
    public int width;
    public int height;

    public byte[] getRgbData() {
        return rgbData;
    }

    public void setRgbData(byte[] rgbData) {
        this.rgbData = rgbData;
    }

    public int getWidth() {
        return width;
    }

    public void setWidth(int width) {
        this.width = width;
    }

    public int getHeight() {
        return height;
    }

    public void setHeight(int height) {
        this.height = height;
    }
}