失眠网 > SpringBoot整合百度云人脸识别功能

SpringBoot整合百度云人脸识别功能

时间：2020-06-06 04:58:37

1.在百度官网创建应用

首先需要在百度智能云官网中创建应用，获取AppID，API Key，Secret Key

官网地址：https://console./

2.添加百度依赖

添加以下依赖即可。其中版本号可在maven官网查询

<dependency><groupId>com.baidu.aip</groupId><artifactId>java-sdk</artifactId><version>${version}</version></dependency>

3.在application.yml中添加属性

便于后面去获取值

4.创建AipFace

AipFace是人脸识别的Java客户端，为使用人脸识别的开发人员提供了一系列的交互方法。初始化完成后建议单例使用，避免重复获取access_token（官方原话）

@Configurationpublic class BaiduConfig {@Value("${baidu.appId}")private String appId;@Value("${baidu.key}")private String key;@Value("${baidu.secret}")private String secret;@Beanpublic AipFace aipFace(){return new AipFace(appId,key,secret);}}

5.注册人脸接口

客户端上传的人脸的图片的Base64编码，并将该用户人脸图形进行本地保存，且存入数据库

@Value("${PhotoPath}")private String filePath;@Autowiredprivate AipFace aipFace;@RequestMapping(value = "register",method = RequestMethod.POST)public String register(String userName,String faceBase) throws IOException {if(!StringUtils.isEmpty(userName) && !StringUtils.isEmpty(faceBase)) {// 文件上传的地址System.out.println(filePath);// 图片名称String fileName = userName + System.currentTimeMillis() + ".png";System.out.println(filePath + "\\" + fileName);File file = new File(filePath + "\\" + fileName);// 往数据库里插入一条用户数据Users user = new Users();user.setUserName(userName);user.setUserPhoto(filePath + "\\" + fileName);//判断用户名是否重复LambdaQueryWrapper<User> queryWrapper = new LambdaQueryWrapper<>();queryWrapper.eq(User::getUserName, userName);User exitUser = userService.getOne(queryWrapper);if (exitUser != null) {return "2";}userService.save(user);// 保存上传摄像头捕获的图片saveLocalImage(faceBase, file);// 向百度云人脸库插入一张人脸faceSetAddUser(aipFace,faceBase,user.getUserId());}return "1";}public boolean saveLocalImage(String imgStr, File file) {// 图像数据为空if (imgStr == null) {return false;}else {BASE64Decoder decoder = new BASE64Decoder();try {// Base64解码byte[] bytes = decoder.decodeBuffer(imgStr);for (int i = 0; i < bytes.length; ++i) {if (bytes[i] < 0) {bytes[i] += 256;}}// 生成jpeg图片if(!file.exists()) {file.getParentFile().mkdir();OutputStream out = new FileOutputStream(file);out.write(bytes);out.flush();out.close();return true;}} catch (Exception e) {e.printStackTrace();return false;}} return false;}public boolean faceSetAddUser(AipFace client, String faceBase, String userId) {// 参数为数据库中注册的人脸HashMap<String, String> options = new HashMap<String, String>();options.put("user_info", "user's info");JSONObject res = client.addUser(faceBase, "BASE64", "user_01", userId, options);return true;}

6.人脸登录接口

@RequestMapping(value = "login",method = RequestMethod.POST)public String login(String faceBase) {String faceData = faceBase;// 进行人像数据对比Double num = verifyUser(faceData,aipFace);if( num > 80) {return "1";}else {return "2";}}public Double verifyUser(String imgBash64,AipFace client) {// 传入可选参数调用接口HashMap<String, String> options = new HashMap<String, String>();JSONObject res = client.search(imgBash64, "BASE64", "user_01", options);JSONObject user = (JSONObject) res.getJSONObject("result").getJSONArray("user_list").get(0);Double score = (Double) user.get("score");return score;}