package com.walker.openocr.support;
|
|
import com.baidu.aip.ocr.AipOcr;
|
import com.fasterxml.jackson.databind.JsonNode;
|
import com.walker.infrastructure.ApplicationRuntimeException;
|
import com.walker.infrastructure.utils.JsonUtils;
|
import com.walker.infrastructure.utils.StringUtils;
|
import com.walker.openocr.AbstractOcrLoader;
|
import com.walker.openocr.TextBlock;
|
import com.walker.openocr.util.BaiduOcrUtils;
|
import org.json.JSONObject;
|
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.Iterator;
|
import java.util.List;
|
|
public class BaiduOcrLoader extends AbstractOcrLoader {
|
|
private HashMap<String, String> options;
|
|
public BaiduOcrLoader(){
|
|
}
|
|
@Override
|
public List<TextBlock> loadText(byte[] image, HashMap<String, String> options) {
|
this.checkOcrClient();
|
JSONObject res = client.general(image, options); // 包含位置
|
if(res.has("error_code")){
|
logger.error("调用返回错误：" + res.getString("error_msg"));
|
return null;
|
}
|
|
// List<TextBlock> textBlocks = new ArrayList<>();
|
// try {
|
// JsonNode jsonNode = JsonUtils.jsonStringToObjectNode(res.toString(2));
|
// JsonNode wordNode = null;
|
// for(Iterator<JsonNode> it = jsonNode.get("words_result").elements(); it.hasNext();){
|
// wordNode = it.next();
|
// textBlocks.add(BaiduOcrUtils.acquireTextBlock(wordNode));
|
// }
|
// } catch (Exception e) {
|
// logger.error("res = {}", res);
|
// throw new ApplicationRuntimeException("百度返回json数据，系统解析错误:" + e.getMessage(), e);
|
// }
|
return this.doLoad(res);
|
}
|
|
@Override
|
public List<TextBlock> loadText(String imagePath, HashMap<String, String> options) {
|
this.checkOcrClient();
|
JSONObject res = client.general(imagePath, options); // 包含位置
|
if(res.has("error_code")){
|
logger.error("调用返回错误：" + res.getString("error_msg"));
|
return null;
|
}
|
return this.doLoad(res);
|
}
|
|
private List<TextBlock> doLoad(JSONObject res){
|
List<TextBlock> textBlocks = new ArrayList<>();
|
try {
|
JsonNode jsonNode = JsonUtils.jsonStringToObjectNode(res.toString(2));
|
JsonNode wordNode = null;
|
for(Iterator<JsonNode> it = jsonNode.get("words_result").elements(); it.hasNext();){
|
wordNode = it.next();
|
textBlocks.add(BaiduOcrUtils.acquireTextBlock(wordNode));
|
}
|
} catch (Exception e) {
|
logger.error("res = {}", res);
|
throw new ApplicationRuntimeException("百度返回json数据，系统解析错误:" + e.getMessage(), e);
|
}
|
return textBlocks;
|
}
|
|
private void checkOcrClient(){
|
if(this.client == null){
|
if(StringUtils.isEmpty(this.appId) || StringUtils.isEmpty(this.apiKey) || StringUtils.isEmpty(this.secretKey)){
|
throw new IllegalArgumentException("appId, apiKey, secretKey为空，无法初始化百度OCR client!");
|
}
|
client = new AipOcr(this.appId, this.apiKey, this.secretKey);
|
// 可选：设置网络连接参数
|
client.setConnectionTimeoutInMillis(2000);
|
client.setSocketTimeoutInMillis(60000);
|
|
// 传入可选参数调用接口
|
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");
|
// 是否返回文字外接多边形顶点位置，不支持单字位置。默认为false
|
// options.put("vertexes_location", "true");
|
logger.info("百度: AipOcr初始化成功");
|
}
|
}
|
|
public String getAppId() {
|
return appId;
|
}
|
|
public void setAppId(String appId) {
|
this.appId = appId;
|
}
|
|
public String getApiKey() {
|
return apiKey;
|
}
|
|
public void setApiKey(String apiKey) {
|
this.apiKey = apiKey;
|
}
|
|
public String getSecretKey() {
|
return secretKey;
|
}
|
|
public void setSecretKey(String secretKey) {
|
this.secretKey = secretKey;
|
}
|
|
private String appId;
|
private String apiKey;
|
private String secretKey;
|
private AipOcr client = null;
|
}
|
