OCR识别行驶证(阿里云和百度云)
一、使用场景
1、通过识别行驶证，获取相关汽车信息，替代手输
二、效果图
三、代码部分：
1、阿里云OCR
1.1、控制层
@PostMapping("/ocrCard")
public JSONObject ocrCard(@RequestPart("file") MultipartFile file, String type){//正面为face，反面为backJSONObject jsonObject = OcrUtils.ocrCard(file,type);return jsonObject;
}
1.2、OcrUtils工具类
public class OcrUtils {public static final String host = "https://driving.market.alicloudapi.com";public static final String path = "/rest/160601/ocr/ocr_vehicle.json";public static final String appcode = "替换成自己申请的";/*** type是前端传值，face为正面，face为反面*/public static JSONObject ocrCard(MultipartFile file,String type){String imgBase64 = encodeImageToBase64(file);JSONObject jsonObject = new JSONObject();//如果文档的输入中含有inputs字段，设置为True， 否则设置为FalseBoolean is_old_format = false;//请根据线上文档修改configure字段JSONObject configObj = new JSONObject();configObj.put("side", type);String config_str = configObj.toString();String method = "POST";Map<String, String> headers = new HashMap<String, String>();//header中的格式headers.put("Authorization", "APPCODE " + appcode);Map<String, String> querys = new HashMap<String, String>();// 拼装请求body的json字符串JSONObject requestObj = new JSONObject();try {if(is_old_format) {JSONObject obj = new JSONObject();obj.put("image", getParam(50, imgBase64));if(config_str.length() > 0) {obj.put("configure", getParam(50, config_str));}JSONArray inputArray = new JSONArray();inputArray.add(obj);requestObj.put("inputs", inputArray);}else{requestObj.put("image", imgBase64);if(config_str.length() > 0) {requestObj.put("configure", config_str);}}} catch (JSONException e) {e.printStackTrace();}String bodys = requestObj.toString();try {HttpResponse response = HttpUtils.doPost(host, path, method, headers, querys, bodys);int stat = response.getStatusLine().getStatusCode();if(stat != 200){System.out.println("Http code: " + stat);System.out.println("http header error msg: "+ response.getFirstHeader("X-Ca-Error-Message"));System.out.println("Http body error msg:" + EntityUtils.toString(response.getEntity()));return jsonObject;}String res = EntityUtils.toString(response.getEntity());JSONObject res_obj = JSON.parseObject(res);if(is_old_format) {JSONArray outputArray = res_obj.getJSONArray("outputs");String output = outputArray.getJSONObject(0).getJSONObject("outputValue").getString("dataValue");JSONObject out = JSON.parseObject(output);return out;}else{return res_obj;}} catch (Exception e) {e.printStackTrace();}return jsonObject;}/** 获取参数的json对象*/public static JSONObject getParam(int type, String dataValue) {JSONObject obj = new JSONObject();try {obj.put("dataType", type);obj.put("dataValue", dataValue);} catch (JSONException e) {e.printStackTrace();}return obj;}private static String encodeImageToBase64(MultipartFile file) {try {byte[] content = file.getBytes();return new String(encodeBase64(content));} catch (IOException e) {e.printStackTrace();return "";}}
}
2、百度云OCR
2.1、控制层
@PostMapping("/ocrCardBd")
public JSONObject ocrCardBd(@RequestPart("file") MultipartFile file,String type){JSONObject jsonObject = OcrBdUtils.ocrCard(file,type);return jsonObject;
}
2.2、OcrBdUtils工具类
public class OcrBdUtils {public static final String apiKey = "替换成自己的";public static final String secretKey = "替换成自己的";public static final String tokenUrl = "https://aip.baidubce.com/oauth/2.0/token";public static final String apiUrl = "https://aip.baidubce.com/rest/2.0/ocr/v1/vehicle_license";/*** type是前端传值，front为正面(默认)，back为反面*/public static JSONObject ocrCard(MultipartFile file, String type) {try {String accessToken = getAccessToken();String apiEndpoint = apiUrl + "?access_token=" + accessToken;// 获取文件字节数组并进行Base64编码byte[] fileBytes = file.getBytes();String imgStr = Base64Util.encode(fileBytes);String imgParam = URLEncoder.encode(imgStr, "UTF-8");// 构造请求的URL和参数URL url = new URL(apiEndpoint);HttpURLConnection connection = (HttpURLConnection) url.openConnection();connection.setRequestMethod("POST");connection.setDoOutput(true);connection.setRequestProperty("Content-Type", "application/x-www-form-urlencoded");// 发送请求参数String postData = "image=" + imgParam + "&vehicle_license_side=" + type;connection.getOutputStream().write(postData.getBytes());// 获取响应InputStream inputStream = connection.getInputStream();String response = readStream(inputStream);return JSONObject.parseObject(response);} catch (IOException e) {e.printStackTrace();return new JSONObject();}}// 获取令牌public static String getAccessToken() throws IOException {URL url = new URL(tokenUrl + "?grant_type=client_credentials&client_id=" + apiKey + "&client_secret=" + secretKey);HttpURLConnection connection = (HttpURLConnection) url.openConnection();connection.setRequestMethod("GET");InputStream responseStream = connection.getInputStream();String response = StreamUtils.copyToString(responseStream, java.nio.charset.StandardCharsets.UTF_8);JSONObject jsonObject = JSONObject.parseObject(response);return jsonObject.getString("access_token");}// 读取输入流的内容private static String readStream(InputStream inputStream) throws IOException {try (ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {byte[] buffer = new byte[1024];int bytesRead;while ((bytesRead = inputStream.read(buffer)) != -1) {outputStream.write(buffer, 0, bytesRead);}return outputStream.toString();}}
}
一个在学习中的开发者，勿喷，欢迎交流