SpringBoot集成百度AI實現(xiàn)人臉識別的項目實踐

更新時間：2022年05月26日 10:29:44 作者：止步前行

本文主要介紹了SpringBoot集成百度AI實現(xiàn)人臉識別的項目實踐，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

1. 百度AI開放平臺

選擇百度AI當(dāng)然就要去創(chuàng)建對于的賬號，同時創(chuàng)建應(yīng)用信息。

進(jìn)入百度AI官網(wǎng)，然后在開放能力Tab里面選擇人臉與人體，按圖中所示操作吧。

該登錄就登錄吧，百度賬號。

對于新用戶來說，可以選擇免費嘗鮮，要不然得充值，練手的話，這個就夠了，反正各種免費領(lǐng)取。

領(lǐng)完后，需要創(chuàng)建測試的應(yīng)用，生成很重要的三個信息：AppID、API Key、Secret Key

2. 文檔集成

創(chuàng)建完應(yīng)用信息后，就可以進(jìn)行SDK集成了，里面有各種語言：Android、IOS、Java、PHP等等，本文集成Java HTTP SDK。

文檔里面說的其實很詳細(xì)了。

3. 代碼實現(xiàn)

框架使用如下：SpringBoot + JPA + MySQL。當(dāng)然JPA可以使用Mybatis或者MybatisPlus

3.1 創(chuàng)建SpringBoot工程

3.2 添加百度AI依賴

添加以下依賴即可。其中版本號可在maven官網(wǎng)查詢

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

3.3 創(chuàng)建AipFace

AipFace是人臉識別的Java客戶端，為使用人臉識別的開發(fā)人員提供了一系列的交互方法。初始化完成后建議單例使用，避免重復(fù)獲取access_token（官方原話）

@Configuration
public class BaiduConfig {

    @Value("${baidu.appId}")
    private String appId;

    @Value("${baidu.key}")
    private String key;

    @Value("${baidu.secret}")
    private String secret;

    @Bean
    public AipFace aipFace(){
        return new AipFace(appId,key,secret);
    }
}

3.4 注冊人臉接口

客戶端上傳的人臉的圖片的Base64編碼，并將該用戶人臉圖形進(jìn)行本地保存，且存入數(shù)據(jù)庫

@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);

        // 往數(shù)據(jù)庫里插入一條用戶數(shù)據(jù)
        Users user = new Users();
        user.setUserName(userName);
        user.setUserPhoto(filePath + "\\" + fileName);
        Users exitUser = userService.selectUserByName(user);
        if(exitUser != null) { 
            return "2";
        }
        userService.addUsers(user);

        // 保存上傳攝像頭捕獲的圖片
        saveLocalImage(faceBase, file);
        // 向百度云人臉庫插入一張人臉
        faceSetAddUser(aipFace,faceBase,userName);
    }
    return "1";
}


public boolean saveLocalImage(String imgStr, File file) {
    // 圖像數(shù)據(jù)為空
    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 username) {
    // 參數(shù)為數(shù)據(jù)庫中注冊的人臉
    HashMap<String, String> options = new HashMap<String, String>();
    options.put("user_info", "user's info");
    JSONObject res = client.addUser(faceBase, "BASE64", "user_01", username, options);
    return true;
}

3.5 人臉登錄接口

@RequestMapping(value = "login",method = RequestMethod.POST)
public String login(String faceBase) {
    String faceData = faceBase;
    // 進(jìn)行人像數(shù)據(jù)對比
    Double num = checkUser(faceData,aipFace);
    if( num > 80) {
        return "1";
    }else {
        return "2";
    }
}

public Double checkUser(String imgBash64,AipFace client) {
    // 傳入可選參數(shù)調(diào)用接口
    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;
}

3.6 頁面

其實比較困難的是這個PC端采集用戶人臉圖像，需要調(diào)用PC攝像頭。

<style type="text/css">
	*{margin: 0;padding: 0;}
	html,body{width: 100%;height: 100%;}/**/
	h1{color: #fff;text-align: center;line-height: 80px;}
	.media{width: 450px;height: 300px;line-height: 300px;margin: 40px auto;}
	.btn{width: 250px;height:50px; line-height:50px; margin: 20px auto; text-align: center;}
	#register{width: 200px;height:50px;background-color: skyblue;text-align: center;line-height: 50px;color: #fff;}
	#canvas{display: none;}
	#shuru{width: 250px;height:50px; line-height:50px;background-color: skyblue; margin: 20px auto; text-align: center;}
</style>
</head>
<body>
	<h1>百度云人臉注冊</h1>
	<div id="shuru">用戶名：<input type="text" name="username" id="username"/></div>
	
	<div class="media">
		<video id="video" width="450" height="300" src="" autoplay></video>
		<canvas id="canvas" width="450" height="300"></canvas>
	</div>
	<div class="btn"><button id="register" >確定注冊</button></div>
	<script type="text/javascript" src="js/jquery-3.3.1.js"></script>

	<script type="text/javascript">
	/**調(diào)用攝像頭，獲取媒體視頻流**/
	var video = document.getElementById('video');
	//返回畫布二維畫圖環(huán)境
	var userContext = canvas.getContext("2d");
	var getUserMedia = 
		//瀏覽器兼容,表示在火狐、Google、IE等瀏覽器都可正常支持
		(navigator.getUserMedia || navigator.webkitGetUserMedia || navigator.mozGetUserMedia || navigator.msGetUserMedia)
		//getUserMedia.call(要調(diào)用的對象，約束條件，調(diào)用成功的函數(shù)，調(diào)用失敗的函數(shù))
		getUserMedia.call(navigator,{video: true,audio: false},function(localMediaStream){
			//獲取攝像頭捕捉的視頻流
			video.srcObject=localMediaStream;
		},function(e){
			console.log("獲取攝像頭失?。?！")
		});
	//點擊按鈕注冊事件
	 var btn = document.getElementById("register");
	
	btn.onclick = function () {
		var username = $("#username").val();
		alert($("#username").val());
			if(username != null){
				//點擊按鈕時拿到登陸者面部信息
		           userContext.drawImage(video,0,0,450,300);
		           var userImgSrc = document.getElementById("canvas").toDataURL("img/png");
		           //拿到bash64格式的照片信息
		           var faceBase = userImgSrc.split(",")[1];
		           
		           //ajax異步請求
		           $.ajax({
		        	   url: "register",
		        	   type: "post",
		        	   data: {"faceBase": faceBase,
		        		   "userName": username
		        	   },
		        	   success: function(result){
		        		   if(result === '1'){
		        			   alert("注冊成功！！，點擊確認(rèn)跳轉(zhuǎn)至登錄頁面");
			        		   window.location.href="toLogin";
		        		   }else if(result === '2'){
		        			   alert("您已經(jīng)注冊過啦！！");
		        		   }else{
		        			   alert("系統(tǒng)錯誤??！");
		        		   }
		        	   }
		           })
			}else{
				alert("用戶名不能為空");
			}
       }
	</script>
</body>