package com.javaweb.third.baidu.api; import com.javaweb.third.baidu.util.Base64Util; import com.javaweb.third.baidu.util.FileUtil; import com.javaweb.third.baidu.util.HttpUtil; import java.net.URLEncoder; /** * 文字识别 * @author wujiyue */ public class OcrAPI extends BaiduAi{ private static final String URL_general_basic="https://aip.baidubce.com/rest/2.0/ocr/v1/general_basic";//通用文字识别 private static final String URL_idcard="https://aip.baidubce.com/rest/2.0/ocr/v1/idcard";//身份证字体识别 private static final String URL_bankcard="https://aip.baidubce.com/rest/2.0/ocr/v1/bankcard"; private static final String URL_driving_license="https://aip.baidubce.com/rest/2.0/ocr/v1/driving_license"; private static final String URL_license_plate="https://aip.baidubce.com/rest/2.0/ocr/v1/license_plate";//车牌号识别 private static final String URL_number="https://aip.baidubce.com/rest/2.0/ocr/v1/numbers"; private static final String URL_qrcode="https://aip.baidubce.com/rest/2.0/ocr/v1/qrcode"; private static final String URL_excel_request="https://aip.baidubce.com/rest/2.0/solution/v1/form_ocr/request"; private static final String URL_excel_get_request_result="https://aip.baidubce.com/rest/2.0/solution/v1/form_ocr/get_request_result"; /** * 通用文字识别 * @param path * @return */ public static String general_basic(String path){ try { byte[] imgData = FileUtil.readFileByBytes(path); String imgStr = Base64Util.encode(imgData); String params = URLEncoder.encode("image", "UTF-8") + "=" + URLEncoder.encode(imgStr, "UTF-8"); String result = HttpUtil.post(URL_general_basic, getAccessToken(), params); return result; } catch (Exception e) { e.printStackTrace(); } return null; } //身份证字体识别 public static String idcardOcr(String path,Boolean isBackGround){ try { byte[] imgData = FileUtil.readFileByBytes(path); String imgStr = Base64Util.encode(imgData); // 识别身份证正面id_card_side=front;识别身份证背面id_card_side=back; String params = ""; if(isBackGround){ params += "id_card_side=back&" ; }else{ params += "id_card_side=front&" ; } params += URLEncoder.encode("image", "UTF-8") + "="+ URLEncoder.encode(imgStr, "UTF-8"); String result = HttpUtil.post(URL_idcard, getAccessToken(), params); return result; } catch (Exception e) { e.printStackTrace(); } return null; } //银行卡识别 public static String bankCardOcr(String path){ try { byte[] imgData = FileUtil.readFileByBytes(path); String imgStr = Base64Util.encode(imgData); String params = URLEncoder.encode("image", "UTF-8") + "=" + URLEncoder.encode(imgStr, "UTF-8"); String result = HttpUtil.post(URL_bankcard, getAccessToken(), params); return result; } catch (Exception e) { e.printStackTrace(); } return null; } public static String drivingLicenseOcr(String path){ try { byte[] imgData = FileUtil.readFileByBytes(path); String imgStr = Base64Util.encode(imgData); String params = URLEncoder.encode("image", "UTF-8") + "=" + URLEncoder.encode(imgStr, "UTF-8"); String result = HttpUtil.post(URL_driving_license, getAccessToken(), params); return result; } catch (Exception e) { e.printStackTrace(); } return null; } //车牌号识别 public static String license_plateOcr(String path){ try { byte[] imgData = FileUtil.readFileByBytes(path); String imgStr = Base64Util.encode(imgData); String params = URLEncoder.encode("image", "UTF-8") + "=" + URLEncoder.encode(imgStr, "UTF-8"); String result = HttpUtil.post(URL_license_plate, getAccessToken(), params); return result; } catch (Exception e) { e.printStackTrace(); } return null; } //数字识别(额外申请权限) protected static String numberOcr(String path){ try { byte[] imgData = FileUtil.readFileByBytes(path); String imgStr = Base64Util.encode(imgData); String params = URLEncoder.encode("image", "UTF-8") + "=" + URLEncoder.encode(imgStr, "UTF-8"); String result = HttpUtil.post(URL_number, getAccessToken(), params); return result; } catch (Exception e) { e.printStackTrace(); } return null; } //二维码识别(额外申请权限) protected static String qrcode(String path){ try { byte[] imgData = FileUtil.readFileByBytes(path); String imgStr = Base64Util.encode(imgData); String params = URLEncoder.encode("image", "UTF-8") + "=" + URLEncoder.encode(imgStr, "UTF-8"); String result = HttpUtil.post(URL_qrcode, getAccessToken(), params); return result; } catch (Exception e) { e.printStackTrace(); } return null; } //表格文字识别(异步接口) //1.请求接口 protected static String excel_request(String path){ try { byte[] imgData = FileUtil.readFileByBytes(path); String imgStr = Base64Util.encode(imgData); String params = URLEncoder.encode("image", "UTF-8") + "=" + URLEncoder.encode(imgStr, "UTF-8"); String result = HttpUtil.post(URL_excel_request, getAccessToken(), params); return result; } catch (Exception e) { e.printStackTrace(); } return null; } /** * 2.获取请求结果 * @param request_id * @param resultType excel：返回生成的excel的网络地址 json:返回json数据 * @return */ protected static String excel_get_request_result(String request_id,String resultType){ try { String params = "request_id=" + request_id+"&result_type="+resultType;//result_type默认为excel返回excel生成地址 String result = HttpUtil.post(URL_excel_get_request_result, getAccessToken(), params); return result; } catch (Exception e) { e.printStackTrace(); } return null; } public static void main(String[] args){ /*String path="C:\\Users\\Administrator\\Desktop\\id.jpg"; String s=idcardOcr(path,false); System.out.println(s);*/ /*String path="C:\\Users\\Administrator\\Desktop\\bank2.jpg"; String s=bankCardOcr(path); System.out.println(s);*/ /*String path="C:\\Users\\Administrator\\Desktop\\drive.jpeg"; String s=drivingLicenseOcr(path); System.out.println(s);*/ /*String path="C:\\Users\\Administrator\\Desktop\\plate.png"; String s=license_plateOcr(path); System.out.println(s);*/ /* String path="C:\\Users\\Administrator\\Desktop\\num.jpg"; String s=numberOcr(path); System.out.println(s);*/ /* String path="C:\\Users\\Administrator\\Desktop\\sex.jpg"; String s=qrcode(path); System.out.println(s);*/ // String path="C:\\Users\\Administrator\\Desktop\\excel.jpg"; // String request_id=excel_request(path); // System.out.println(request_id); // String s=excel_get_request_result("11394947_336225","excel"); // System.out.println(s); String path="C:\\Users\\Administrator\\Desktop\\2.jpg"; String res=general_basic(path); System.out.println(res); } }