Browse Source

发票识别ocr

ptos
zhizhi wu 4 years ago
parent
commit
560c860be4
  1. 11
      src/main/java/com/ccsens/ptccsens/service/OcrService.java
  2. 2
      src/main/java/com/ccsens/ptccsens/util/BasicsConstant.java

11
src/main/java/com/ccsens/ptccsens/service/OcrService.java

@ -1,7 +1,9 @@
package com.ccsens.ptccsens.service;
import cn.hutool.core.util.StrUtil;
import com.ccsens.ptccsens.bean.vo.OcrVo;
import com.ccsens.ptccsens.util.BasicsConstant;
import com.ccsens.util.RestTemplateUtil;
import com.ccsens.util.baidu.BaiDuDto;
import com.ccsens.util.baidu.BaiDuUtil;
import com.ccsens.util.baidu.BaiDuVo;
@ -20,8 +22,13 @@ public class OcrService implements IOcrService{
// 图像识别
BaiDuDto.GeneralBasic basic = new BaiDuDto.GeneralBasic();
basic.setImage(img);
BaiDuVo.BillBasic words = BaiDuUtil.billBasic(BasicsConstant.BaiDu.APP_KEY, BasicsConstant.BaiDu.SECRET_KEY, basic);
log.info("识别结果:{}",words);
// BaiDuVo.BillBasic words = BaiDuUtil.billBasic(BasicsConstant.BaiDu.APP_KEY, BasicsConstant.BaiDu.SECRET_KEY, basic);
// log.info("识别结果:{}",words);
String token = BaiDuUtil.getToken(BasicsConstant.BaiDu.INVOICE_APP_KEY, BasicsConstant.BaiDu.INVOICE_SECRET_KEY);
String invoiceUrl = "https://aip.baidubce.com/rest/2.0/ocr/v1/vat_invoice?access_token={}";
String url = StrUtil.format(invoiceUrl, token);
String result = RestTemplateUtil.postUrlEncode(url, basic);
log.info("result:{}", result);
// 返回数据
OcrVo.BillInfo personMsg = new OcrVo.BillInfo();
// personMsg.toMsg(words.getWordsResult());

2
src/main/java/com/ccsens/ptccsens/util/BasicsConstant.java

@ -56,6 +56,8 @@ public class BasicsConstant {
public static class BaiDu{
public static final String APP_KEY = "F43SLi3hDra3EgWiSi8bIH8c";
public static final String SECRET_KEY = "wGhbTXGsrbxfuCQT7WyLDndYRxrSYqbD";
public static final String INVOICE_APP_KEY = "MPTHGR2ytloHo61D1PQipjo7";
public static final String INVOICE_SECRET_KEY = "2rENBsctjjSXbRFhMCXi6CBUOUVvNrSK";
public static final String KEY = "key";
public static final String NUM = "num";
}

Loading…
Cancel
Save