首先去虹软官网注册获取对应的
官网下载对应的 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;}
}