用 Java 实现人脸识别功能

x33g5p2x  于2021-12-02 转载在 Java  
字(8.5k)|赞(0)|评价(0)|浏览(421)

引言

人脸识别SDK

人脸识别技术是很复杂的,自己用Java手撕一个识别算法有点不切实际,毕竟实力不允许我这么嚣张,还是借助三方的SDK吧!

找了一圈发现一个免费的人脸识别SDK: ArcSoft:,地址:https://ai.arcsoft.com.cn

官网首页 -> 右上角开发者中心 -> 选择“人脸识别” -> 添加SDK,会生成APPIDSDK KEY后续会用到,根据需要选择不同的环境(本文基于windows环境),然后下载SDK是一个压缩包。

Java项目搭建

终于在我的苦苦搜寻之下终于,找到一个ArcSoftJava版本Demo,开源真是一件美好的事情,话不多说开干!

1、下载demo项目

github地址:https://github.com/xinzhfiu/ArcSoftFaceDemo,本地搭建数据库,创建表:user_face_info。这个表主要用来存人像特征,其中主要的字段 face_feature 用二进制类型 blob 存放人脸特征。

  1. SET NAMES utf8mb4;
  2. SET FOREIGN_KEY_CHECKS = 0;
  3. -- ----------------------------
  4. -- Table structure for user_face_info
  5. -- ----------------------------
  6. DROP TABLE IF EXISTS `user_face_info`;
  7. CREATE TABLE `user_face_info` (
  8. `id` int(11) NOT NULL AUTO_INCREMENT COMMENT '主键',
  9. `group_id` int(11) DEFAULT NULL COMMENT '分组id',
  10. `face_id` varchar(31) DEFAULT NULL COMMENT '人脸唯一Id',
  11. `name` varchar(63) DEFAULT NULL COMMENT '名字',
  12. `age` int(3) DEFAULT NULL COMMENT '年纪',
  13. `email` varchar(255) DEFAULT NULL COMMENT '邮箱地址',
  14. `gender` smallint(1) DEFAULT NULL COMMENT '性别,1=男,2=女',
  15. `phone_number` varchar(11) DEFAULT NULL COMMENT '电话号码',
  16. `face_feature` blob COMMENT '人脸特征',
  17. `create_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
  18. `update_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间',
  19. `fpath` varchar(255) COMMENT '照片路径',
  20. PRIMARY KEY (`id`) USING BTREE,
  21. KEY `GROUP_ID` (`group_id`) USING BTREE
  22. ) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8mb4 ROW_FORMAT=DYNAMIC;
  23. SET FOREIGN_KEY_CHECKS = 1;

2、修改application.properties文件

整个项目还是比较完整的,只需改一些配置即可启动,但有几点注意的地方,后边会重点说明。

config.arcface-sdk.sdk-lib-path: 存放SDK压缩包中的三个.dll文件的路径

  1. config.arcface-sdk.sdk-lib-path=d:/arcsoft_lib
  2. config.arcface-sdk.app-id=8XMHMu71Dmb5UtAEBpPTB1E9ZPNTw2nrvQ5bXxBobUA8
  3. config.arcface-sdk.sdk-key=BA8TLA9vVwK7G6btJh2A2FCa8ZrC6VWZLNbBBFctCz5R
  4. # druid 本地的数据库地址
  5. spring.datasource.druid.url=jdbc:mysql://127.0.0.1:3306/xin-master?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=UTC
  6. spring.datasource.druid.username=junkang
  7. spring.datasource.druid.password=junkang

3、根目录创建lib文件夹

在项目根目录创建文件夹 lib,将下载的SDK压缩包中的arcsoft-sdk-face-2.2.0.1.jar放入项目根目录

4、引入arcsoft依赖包

  1. <dependency>
  2. <groupId>com.arcsoft.face</groupId>
  3. <artifactId>arcsoft-sdk-face</artifactId>
  4. <version>2.2.0.1</version>
  5. <scope>system</scope>
  6. <systemPath>${basedir}/lib/arcsoft-sdk-face-2.2.0.1.jar</systemPath>
  7. </dependency>

pom.xml文件要配置includeSystemScope属性,否则可能会导致arcsoft-sdk-face-2.2.0.1.jar引用不到

  1. <build>
  2. <plugins>
  3. <plugin>
  4. <groupId>org.springframework.boot</groupId>
  5. <artifactId>spring-boot-maven-plugin</artifactId>
  6. <configuration>
  7. <includeSystemScope>true</includeSystemScope>
  8. <fork>true</fork>
  9. </configuration>
  10. </plugin>
  11. </plugins>
  12. </build>

5、启动项目

到此为止配置完成,run``Application文件启动

测试一下:http://127.0.0.1:8089/demo,如下页面即启动成功

操作

1、录入人脸图像

页面输入名称,点击摄像头注册调起本地摄像头,提交后将当前图像传入后台,识别提取当前人脸体征,保存至数据库。

2、人脸对比

录入完人脸图像后测试一下能否识别成功,提交当前的图像,发现识别成功相似度92%。但是作为程序员对什么事情都要持怀疑的态度,这结果不是老铁在页面写死的吧?

为了进一步验证,这回把脸挡住再试一下,发现提示“人脸不匹配”,证明真的有进行比对。

源码分析

简单看了一下项目源码,分析一下实现的过程:

页面和JS一看就是后端程序员写的,不要问我问为什么?懂的自然懂,哈哈哈 ~ ,

1、JS调起本地摄像头拍照,上传图片文件字符串

  1. function getMedia() {
  2. $("#mainDiv").empty();
  3. let videoComp = " <video id='video' width='500px' height='500px' autoplay='autoplay' style='margin-top: 20px'></video><canvas id='canvas' width='500px' height='500px' style='display: none'></canvas>";
  4. $("#mainDiv").append(videoComp);
  5. let constraints = {
  6. video: {width: 500, height: 500},
  7. audio: true
  8. };
  9. //获得video摄像头区域
  10. let video = document.getElementById("video");
  11. //这里介绍新的方法,返回一个 Promise对象
  12. // 这个Promise对象返回成功后的回调函数带一个 MediaStream 对象作为其参数
  13. // then()是Promise对象里的方法
  14. // then()方法是异步执行,当then()前的方法执行完后再执行then()内部的程序
  15. // 避免数据没有获取到
  16. let promise = navigator.mediaDevices.getUserMedia(constraints);
  17. promise.then(function (MediaStream) {
  18. video.srcObject = MediaStream;
  19. video.play();
  20. });
  21. // var t1 = window.setTimeout(function() {
  22. // takePhoto();
  23. // },2000)
  24. }
  25. //拍照事件
  26. function takePhoto() {
  27. let mainComp = $("#mainDiv");
  28. if(mainComp.has('video').length)
  29. {
  30. let userNameInput = $("#userName").val();
  31. if(userNameInput == "")
  32. {
  33. alert("姓名不能为空!");
  34. return false;
  35. }
  36. //获得Canvas对象
  37. let video = document.getElementById("video");
  38. let canvas = document.getElementById("canvas");
  39. let ctx = canvas.getContext('2d');
  40. ctx.drawImage(video, 0, 0, 500, 500);
  41. var formData = new FormData();
  42. var base64File = canvas.toDataURL();
  43. var userName = $("#userName").val();
  44. formData.append("file", base64File);
  45. formData.append("name", userName);
  46. formData.append("groupId", "101");
  47. $.ajax({
  48. type: "post",
  49. url: "/faceAdd",
  50. data: formData,
  51. contentType: false,
  52. processData: false,
  53. async: false,
  54. success: function (text) {
  55. var res = JSON.stringify(text)
  56. if (text.code == 0) {
  57. alert("注册成功")
  58. } else {
  59. alert(text.message)
  60. }
  61. },
  62. error: function (error) {
  63. alert(JSON.stringify(error))
  64. }
  65. });
  66. }
  67. else{
  68. var formData = new FormData();
  69. let userName = $("#userName").val();
  70. formData.append("groupId", "101");
  71. var file = $("#file0")[0].files[0];
  72. var reader = new FileReader();
  73. reader.readAsDataURL(file);
  74. reader.onload = function () {
  75. var base64 = reader.result;
  76. formData.append("file", base64);
  77. formData.append("name",userName);
  78. $.ajax({
  79. type: "post",
  80. url: "/faceAdd",
  81. data: formData,
  82. contentType: false,
  83. processData: false,
  84. async: false,
  85. success: function (text) {
  86. var res = JSON.stringify(text)
  87. if (text.code == 0) {
  88. alert("注册成功")
  89. } else {
  90. alert(text.message)
  91. }
  92. },
  93. error: function (error) {
  94. alert(JSON.stringify(error))
  95. }
  96. });
  97. location.reload();
  98. }
  99. }
  100. }

2、后台解析图片,提取人像特征

后台解析前端传过来的图片,提取人像特征存入数据库,人像特征的提取主要是靠FaceEngine引擎,顺着源码一路看下去,自己才疏学浅实在是没懂具体是个什么样的算法。

  1. /* 人脸添加 */
  2. @RequestMapping(value = "/faceAdd", method = RequestMethod.POST)
  3. @ResponseBody
  4. public Result<Object> faceAdd(@RequestParam("file") String file, @RequestParam("groupId") Integer groupId, @RequestParam("name") String name) {
  5. try {
  6. //解析图片
  7. byte[] decode = Base64.decode(base64Process(file));
  8. ImageInfo imageInfo = ImageFactory.getRGBData(decode);
  9. //人脸特征获取
  10. byte[] bytes = faceEngineService.extractFaceFeature(imageInfo);
  11. if (bytes == null) {
  12. return Results.newFailedResult(ErrorCodeEnum.NO_FACE_DETECTED);
  13. }
  14. UserFaceInfo userFaceInfo = new UserFaceInfo();
  15. userFaceInfo.setName(name);
  16. userFaceInfo.setGroupId(groupId);
  17. userFaceInfo.setFaceFeature(bytes);
  18. userFaceInfo.setFaceId(RandomUtil.randomString(10));
  19. //人脸特征插入到数据库
  20. userFaceInfoService.insertSelective(userFaceInfo);
  21. logger.info("faceAdd:" + name);
  22. return Results.newSuccessResult("");
  23. } catch (Exception e) {
  24. logger.error("", e);
  25. }
  26. return Results.newFailedResult(ErrorCodeEnum.UNKNOWN);
  27. }

3、人像特征对比

人脸识别:将前端传入的图像经过人像特征提取后,和库中已存在的人像信息对比分析

  1. /* 人脸识别 */
  2. @RequestMapping(value = "/faceSearch", method = RequestMethod.POST)
  3. @ResponseBody
  4. public Result<FaceSearchResDto> faceSearch(String file, Integer groupId) throws Exception {
  5. byte[] decode = Base64.decode(base64Process(file));
  6. BufferedImage bufImage = ImageIO.read(new ByteArrayInputStream(decode));
  7. ImageInfo imageInfo = ImageFactory.bufferedImage2ImageInfo(bufImage);
  8. //人脸特征获取
  9. byte[] bytes = faceEngineService.extractFaceFeature(imageInfo);
  10. if (bytes == null) {
  11. return Results.newFailedResult(ErrorCodeEnum.NO_FACE_DETECTED);
  12. }
  13. //人脸比对,获取比对结果
  14. List<FaceUserInfo> userFaceInfoList = faceEngineService.compareFaceFeature(bytes, groupId);
  15. if (CollectionUtil.isNotEmpty(userFaceInfoList)) {
  16. FaceUserInfo faceUserInfo = userFaceInfoList.get(0);
  17. FaceSearchResDto faceSearchResDto = new FaceSearchResDto();
  18. BeanUtil.copyProperties(faceUserInfo, faceSearchResDto);
  19. List<ProcessInfo> processInfoList = faceEngineService.process(imageInfo);
  20. if (CollectionUtil.isNotEmpty(processInfoList)) {
  21. //人脸检测
  22. List<FaceInfo> faceInfoList = faceEngineService.detectFaces(imageInfo);
  23. int left = faceInfoList.get(0).getRect().getLeft();
  24. int top = faceInfoList.get(0).getRect().getTop();
  25. int width = faceInfoList.get(0).getRect().getRight() - left;
  26. int height = faceInfoList.get(0).getRect().getBottom() - top;
  27. Graphics2D graphics2D = bufImage.createGraphics();
  28. graphics2D.setColor(Color.RED);//红色
  29. BasicStroke stroke = new BasicStroke(5f);
  30. graphics2D.setStroke(stroke);
  31. graphics2D.drawRect(left, top, width, height);
  32. ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
  33. ImageIO.write(bufImage, "jpg", outputStream);
  34. byte[] bytes1 = outputStream.toByteArray();
  35. faceSearchResDto.setImage("data:image/jpeg;base64," + Base64Utils.encodeToString(bytes1));
  36. faceSearchResDto.setAge(processInfoList.get(0).getAge());
  37. faceSearchResDto.setGender(processInfoList.get(0).getGender().equals(1) ? "女" : "男");
  38. }
  39. return Results.newSuccessResult(faceSearchResDto);
  40. }
  41. return Results.newFailedResult(ErrorCodeEnum.FACE_DOES_NOT_MATCH);
  42. }

整个人脸识别功能的大致流程图如下:

总结

整个项目的设计思路比较清晰,难点在于人脸识别引擎前端JS部分代码,其他的功能比较平常。

相关文章

最新文章

更多