【百度OCR 封装篇】OCR封装只IOCR自定义模版或分类器封装两种调用方式

客官°小女子只卖身不卖艺 2023-05-31 08:00 4阅读 0赞

全部代码下载地址：[https://download.csdn.net/download/qq\_33333654/11914449][https_download.csdn.net_download_qq_33333654_11914449]

>     package com.jeeplus.ocr;
>     
>     import com.alibaba.fastjson.JSON;
>     import com.baidu.aip.ocr.AipOcr;
>     import com.baidu.aip.util.Base64Util;
>     import com.jeeplus.common.utils.CacheUtils;
>     import com.jeeplus.common.utils.DateUtils;
>     import com.jeeplus.common.utils.HttpUtil;
>     import com.jeeplus.common.utils.StringUtils;
>     import com.jeeplus.modules.sys.common.Constants;
>     import com.jeeplus.ocr.bean.AccessTokenResultBean;
>     import com.jeeplus.ocr.bean.AipOcrBasicGeneralResultBean;
>     import com.jeeplus.ocr.bean.AipOcrCustomResultBean;
>     import org.json.JSONObject;
>     
>     import java.io.*;
>     import java.net.HttpURLConnection;
>     import java.net.URL;
>     import java.net.URLEncoder;
>     import java.util.*;
>     
>     /**
>      * @ProjectName: Hematopathy
>      * @Package: com.jeeplus.ocr
>      * @ClassName: AipOcrUtil
>      * @Author: MC
>      * @Description: ${description}
>      * @Date: 2019/10/23 0023 10:48
>      * @Version: 1.0
>      */
>     public class AipOcrUtil {
>     
>         /**
>          * @Method 通用文字识别
>          * @Author MC
>          path:本地文件路径
>         options:请求API参数，可为空
>          * @Return
>          * @Date 2019/10/23 0023 11:36
>          */
>         public static AipOcrBasicGeneralResultBean basicGeneralByPath(String path, HashMap<String, String> options){
>             if(checkFileIsImg(path)){
>                 if(options == null || options.isEmpty()){
>                     options = new HashMap<String, String>();
>                 }
>                 options.put("language_type", "CHN_ENG");
>                 options.put("detect_direction", "true");
>                 options.put("detect_language", "true");
>                 options.put("probability", "true");
>                 AipOcr client = AipOcrSingleton.getAipOcr();
>                 JSONObject res = client.basicGeneral(path, options);
>                 String resStr = res.toString();
>                 AipOcrBasicGeneralResultBean bean = JSON.parseObject(resStr, AipOcrBasicGeneralResultBean.class);
>                 return bean;
>             }
>             return null;
>         }
>     
>         /**
>          * @Method 通用文字识别
>          * @Author MC
>         url:网络图片地址
>         options:请求API参数，可为空
>          * @Return
>          * @Date 2019/10/23 0023 11:36
>          */
>         public static AipOcrBasicGeneralResultBean basicGeneralByURL(String url, HashMap<String, String> options){
>             if(!StringUtils.isEmpty(url)){
>                 if(options == null || options.isEmpty()){
>                     options = new HashMap<String, String>();
>                 }
>                 options.put("language_type", "CHN_ENG");
>                 options.put("detect_direction", "true");
>                 options.put("detect_language", "true");
>                 options.put("probability", "true");
>                 AipOcr client = AipOcrSingleton.getAipOcr();
>                 JSONObject res = client.basicGeneralUrl(url, options);
>                 String resStr = res.toString();
>                 AipOcrBasicGeneralResultBean bean = JSON.parseObject(resStr, AipOcrBasicGeneralResultBean.class);
>                 return bean;
>             }
>             return null;
>         }
>     
>     
>         /**
>          * @Method 根据模版ID或者分类器ID识别本地图片
>          * @Author MC
>         path：本地图片路径
>         templateSign：指定模版ID
>         classifierId：指定分类器ID
>         模版ID与分类ID至少一个不为空
>          * @Return
>          * @Date 2019/10/23 0023 13:59
>          */
>         public static AipOcrCustomResultBean customByPath(String path,String templateSign,String classifierId){
>             if(checkFileIsImg(path)){
>                 if(StringUtils.isEmpty(templateSign) && StringUtils.isEmpty(classifierId)){
>                     return null;
>                 }
>                 HashMap<String, String> options = new HashMap<String, String>();
>                 if(!StringUtils.isEmpty(templateSign)){
>                     options.put("templateSign", templateSign);
>                 }
>                 if(!StringUtils.isEmpty(classifierId)){
>                     options.put("classifierId", classifierId);
>                 }
>                 AipOcr client = AipOcrSingleton.getAipOcr();
>                 JSONObject res = client.custom(path, options);
>                 String resStr = res.toString();
>                 AipOcrCustomResultBean bean = JSON.parseObject(resStr, AipOcrCustomResultBean.class);
>                 return bean;
>             }
>             return null;
>         }
>     
>         /**
>          * @Method 使用API调用方式进行自定义模版或者分类器的文字识别
>          * @Author MC
>          ak:app_key 必传
>         sk:secret_key 必传
>         filePath:本地文件路径 要求为图片格式 必传
>         templateSign：自定义模版ID
>         classifierParams：自定义分类器ID
>         templateSign和classifierParams至少传递一个
>          * @Return
>          * @Date 2019/10/24 0024 11:34
>          */
>         public static AipOcrCustomResultBean customByApiUrl(String ak, String sk,String filePath,String templateSign,String classifierParams) throws Exception {
>             if(StringUtils.isEmpty(ak) || StringUtils.isEmpty(sk) || StringUtils.isEmpty(filePath)){
>                 return null;
>             }
>             if(StringUtils.isEmpty(templateSign) && StringUtils.isEmpty(classifierParams)){
>                 return null;
>             }
>             if(checkFileIsImg(filePath)){
>                 String imgStr = imgFile2Bytes2Encode(filePath);
>                 String param = "";
>                 if(!StringUtils.isEmpty(classifierParams)){
>                     param = "classifierId="+classifierParams+"&image=" + URLEncoder.encode(imgStr, "UTF-8");
>                 }else{
>                     param = "templateSign="+templateSign+"&image=" + URLEncoder.encode(imgStr, "UTF-8");
>                 }
>                 AccessTokenResultBean auth = getAuth(ak, sk);
>                 if(auth == null || !StringUtils.isEmpty(auth.getError())){
>                     return null;
>                 }
>     
>                 String accessToken = auth.getAccess_token();
>                 // 请求模板识别
>                 String result = HttpUtil.ocrPost(Constants.OCR_API_URL_GET_RECOGNISE, accessToken, param);
>                 AipOcrCustomResultBean bean = JSON.parseObject(result, AipOcrCustomResultBean.class);
>                 return bean;
>             }
>             return null;
>         }
>     
>         /**
>          * 获取API访问token
>          * 该token有一定的有效期，需要自行管理，当失效时需重新获取.
>          * @param ak - 百度云官网获取的 API Key
>          * @param sk - 百度云官网获取的 Securet Key
>          * @return assess_token 示例：
>          * "24.460da4889caad24cccdb1fea17221975.2592000.1491995545.282335-1234567"
>          */
>         public static AccessTokenResultBean getAuth(String ak, String sk) {
>             AccessTokenResultBean bean = (AccessTokenResultBean) CacheUtils.get(Constants.OCR_CACHE_NAME,Constants.OCR_CACHE_ACCESS_TOKEN);
>             if(bean != null){
>                 //检查缓存中的token是否过期
>                 long expires_in = bean.getMyTimeOut().getTime();
>                 Date nowDate = new Date();
>                 long nowDateTime = nowDate.getTime();
>                 if(expires_in > nowDateTime){
>                     return bean;
>                 }else{
>                     CacheUtils.remove(Constants.OCR_CACHE_NAME,Constants.OCR_CACHE_ACCESS_TOKEN);
>                 }
>             }
>     
>             // 获取token地址
>             String authHost = Constants.OCR_API_URL_GET_ACCESS_TOKEN + "?";
>             String getAccessTokenUrl = authHost
>                     // 1. grant_type为固定参数
>                     + "grant_type=client_credentials"
>                     // 2. 官网获取的 API Key
>                     + "&client_id=" + ak
>                     // 3. 官网获取的 Secret Key
>                     + "&client_secret=" + sk;
>             try {
>                 URL realUrl = new URL(getAccessTokenUrl);
>                 // 打开和URL之间的连接
>                 HttpURLConnection connection = (HttpURLConnection) realUrl.openConnection();
>                 connection.setRequestMethod("GET");
>                 connection.connect();
>                 // 获取所有响应头字段
>                 Map<String, List<String>> map = connection.getHeaderFields();
>                 // 遍历所有的响应头字段
>     //            for (String key : map.keySet()) {
>     //                System.err.println(key + "--->" + map.get(key));
>     //            }
>                 // 定义 BufferedReader输入流来读取URL的响应
>                 BufferedReader in = new BufferedReader(new InputStreamReader(connection.getInputStream()));
>                 String result = "";
>                 String line;
>                 while ((line = in.readLine()) != null) {
>                     result += line;
>                 }
>                 /**
>                  * 返回结果示例
>                  */
>                 System.err.println("result:" + result);
>                 AccessTokenResultBean accessTokenResultBean = JSON.parseObject(result, AccessTokenResultBean.class);
>                 if(StringUtils.isEmpty(accessTokenResultBean.getError())){//获取成功 进行缓存
>                     Integer expires_in = accessTokenResultBean.getExpires_in();
>                     if(expires_in > 0){
>                         int days = expires_in / ( 60 * 60 * 24);
>                         Date date=new Date();
>                         Calendar calendar = new GregorianCalendar();
>                         calendar.setTime(date);
>                         calendar.add(Calendar.DATE,days);//把日期往后增加一天.整数往后推,负数往前移动
>                         date=calendar.getTime();
>                         accessTokenResultBean.setMyTimeOut(date);//设置有效期
>                     }else{
>                         CacheUtils.remove(Constants.OCR_CACHE_NAME,Constants.OCR_CACHE_ACCESS_TOKEN);
>                         return null;
>                     }
>                     CacheUtils.put(Constants.OCR_CACHE_NAME,Constants.OCR_CACHE_ACCESS_TOKEN,accessTokenResultBean);
>                 }else{
>                     CacheUtils.remove(Constants.OCR_CACHE_NAME,Constants.OCR_CACHE_ACCESS_TOKEN);
>                 }
>                 return accessTokenResultBean;
>                /* JSONObject jsonObject = new JSONObject(result);
>                 String access_token = jsonObject.getString("access_token");
>                 return access_token;*/
>             } catch (Exception e) {
>                 System.err.printf("获取token失败！");
>                 CacheUtils.remove(Constants.OCR_CACHE_NAME,Constants.OCR_CACHE_ACCESS_TOKEN);
>                 e.printStackTrace(System.err);
>             }
>             return null;
>         }
>     
>         private static boolean checkFileIsImg(String path){
>             if(StringUtils.isEmpty(path)){
>                 return false;
>             }
>             File srcFile = new File(path);
>             // 判断源文件是否存在
>             if (!srcFile.exists()) {
>                 return false;
>             }
>             if(path.indexOf(".") == -1){
>                 return false;
>             }
>             String suffix= path.substring(path.indexOf("."), path.length());
>             if(
>                     suffix.equalsIgnoreCase("jpg") ||
>                             suffix.equalsIgnoreCase("PNG ") ||
>                             suffix.equalsIgnoreCase("GIF") ||
>                             suffix.equalsIgnoreCase("tif") ||
>                             suffix.equalsIgnoreCase("bmp")
>             ){
>                 return false;
>             }
>     
>             return true;
>         }
>     
>     
>         /**
>          * @Method 图片文件转换为字节在进行base64转码
>          * @Author MC
>     
>          * @Return
>          * @Date 2019/10/24 0024 10:19
>          */
>         public static String imgFile2Bytes2Encode(String filePath) throws IOException {
>             if(!checkFileIsImg(filePath)){
>                 return null;
>             }
>             byte[] imgData = readFileByBytes(filePath);
>             String imgStr = Base64Util.encode(imgData);
>             return imgStr;
>         }
>     
>         /**
>          * @Method 文件转换为字节在进行base64转码
>          * @Author MC
>     
>          * @Return
>          * @Date 2019/10/24 0024 10:19
>          */
>         public static String file2Bytes2Encode(String filePath) throws IOException {
>             if(StringUtils.isEmpty(filePath)){
>                 return null;
>             }
>             File srcFile = new File(filePath);
>             // 判断源文件是否存在
>             if (!srcFile.exists()) {
>                 return null;
>             }
>             byte[] imgData = readFileByBytes(filePath);
>             String imgStr = Base64Util.encode(imgData);
>             return imgStr;
>         }
>     
>         /**
>          * 根据文件路径读取byte[] 数组
>          */
>         public static byte[] readFileByBytes(String filePath) throws IOException {
>             File file = new File(filePath);
>             if (!file.exists()) {
>                 throw new FileNotFoundException(filePath);
>             } else {
>                 ByteArrayOutputStream bos = new ByteArrayOutputStream((int) file.length());
>                 BufferedInputStream in = null;
>     
>                 try {
>                     in = new BufferedInputStream(new FileInputStream(file));
>                     short bufSize = 1024;
>                     byte[] buffer = new byte[bufSize];
>                     int len1;
>                     while (-1 != (len1 = in.read(buffer, 0, bufSize))) {
>                         bos.write(buffer, 0, len1);
>                     }
>     
>                     byte[] var7 = bos.toByteArray();
>                     return var7;
>                 } finally {
>                     try {
>                         if (in != null) {
>                             in.close();
>                         }
>                     } catch (IOException var14) {
>                         var14.printStackTrace();
>                     }
>     
>                     bos.close();
>                 }
>             }
>         }
>     
>     
>     
>     
>     }

[https_download.csdn.net_download_qq_33333654_11914449]: https://download.csdn.net/download/qq_33333654/11914449