master
周文涛 2 years ago
parent bfe706f970
commit 33d29b3232

@ -47,6 +47,6 @@ public class OcrConstant {
//目前仅支持的字段 //目前仅支持的字段
public static List<String> SUPPORTED_FIELDS= Arrays.asList("hospitalName","doctorName","departmentName","time"); public static List<String> SUPPORTED_FIELDS= Arrays.asList("hospitalName","doctorName","departmentName","time");
//目前仅支持的字段中文名 //目前仅支持的字段中文名
public static List<String> SUPPORTED_FIELD_NAMES= Arrays.asList("医院名称","医生名称","姓名","科室","时间"); public static List<String> SUPPORTED_FIELD_NAMES= Arrays.asList("医院名称","医生名称","姓名","科室","时间","日期");
} }

@ -253,6 +253,7 @@ public class OcrMetadataConfigController extends JeecgController<OcrMetadataConf
List<String> getFieldList; List<String> getFieldList;
for (OcrMetadataConfigDTO configVo : list) { for (OcrMetadataConfigDTO configVo : list) {
saveOcrMetadataConfigVO =new SaveOcrMetadataConfigVO(); saveOcrMetadataConfigVO =new SaveOcrMetadataConfigVO();
AssertUtils.notEmpty(configVo.getConfigName(),"[配置名称]不可为空");
saveOcrMetadataConfigVO.setConfigName(configVo.getConfigName()); saveOcrMetadataConfigVO.setConfigName(configVo.getConfigName());
resultName = configVo.getResultName(); resultName = configVo.getResultName();
getField = configVo.getGetField(); getField = configVo.getGetField();

@ -184,7 +184,6 @@ public class HandleTransInit implements ApplicationRunner {
}else{ }else{
log.info("图片不存在:"); log.info("图片不存在:");
log.info(masterTaskId); log.info(masterTaskId);
//log.info(semanticResponseJson.toJSONString());
} }
//该子任务已执行,判断主任务是否残留 //该子任务已执行,判断主任务是否残留
String masterTask = (String) redisUtil.get("identify_" + masterTaskId); String masterTask = (String) redisUtil.get("identify_" + masterTaskId);

@ -202,6 +202,14 @@ public class OcrIdentifyServiceImpl extends ServiceImpl<OcrIdentifyMapper, OcrId
break; break;
} }
} }
}else if (Arrays.asList("日期", "时间").contains(s)) {
for (String s1 : Arrays.asList("日期", "时间")) {
JSONArray semantic_result = semanticResult.getJSONObject("semantic_result").getJSONArray(s1);
if (CollectionUtils.isNotEmpty(semantic_result)) {
jsonArray.addAll(semantic_result);
break;
}
}
}else{ }else{
jsonArray = semanticResult.getJSONArray(s); jsonArray = semanticResult.getJSONArray(s);
} }
@ -593,7 +601,7 @@ public class OcrIdentifyServiceImpl extends ServiceImpl<OcrIdentifyMapper, OcrId
e.printStackTrace(); e.printStackTrace();
} }
String task_ = (String) redisUtil.get("task_identify_" + taskLevel);//TODO task_ String task_ = (String) redisUtil.get("task_identify_" + taskLevel);//TODO task_
log.info("历史任务:" + task_); //log.info("历史任务:" + task_);
if (task_ == null || task_.equals("[]")) { if (task_ == null || task_.equals("[]")) {
//无历史任务 //无历史任务
TaskModel task = new TaskModel(taskId, taskLevel, taskType, "0", parameter); TaskModel task = new TaskModel(taskId, taskLevel, taskType, "0", parameter);
@ -778,7 +786,15 @@ public class OcrIdentifyServiceImpl extends ServiceImpl<OcrIdentifyMapper, OcrId
} }
} }
}else if (Arrays.asList("医院名称","医院").contains(s)) { }else if (Arrays.asList("医院名称","医院").contains(s)) {
for (String s1 : Arrays.asList("医院名称","医院")) { for (String s1 : Arrays.asList("医院名称", "医院")) {
JSONArray semantic_result = semanticResult.getJSONObject("semantic_result").getJSONArray(s1);
if (CollectionUtils.isNotEmpty(semantic_result)) {
jsonArray.addAll(semantic_result);
break;
}
}
}else if (Arrays.asList("日期", "时间").contains(s)) {
for (String s1 : Arrays.asList("日期", "时间")) {
JSONArray semantic_result = semanticResult.getJSONObject("semantic_result").getJSONArray(s1); JSONArray semantic_result = semanticResult.getJSONObject("semantic_result").getJSONArray(s1);
if (CollectionUtils.isNotEmpty(semantic_result)) { if (CollectionUtils.isNotEmpty(semantic_result)) {
jsonArray.addAll(semantic_result); jsonArray.addAll(semantic_result);

Loading…
Cancel
Save