package com.iplatform.openocr;
|
|
import com.walker.infrastructure.utils.Base64;
|
import com.walker.infrastructure.utils.FileUtils;
|
import com.walker.openocr.OcrType;
|
import com.walker.openocr.TextBlock;
|
import com.walker.openocr.idcard.IdCardConfig;
|
import com.walker.openocr.idcard.IdCardObject;
|
import com.walker.openocr.idcard.IdCardTextResolver;
|
import com.walker.openocr.table.CellObject;
|
import com.walker.openocr.table.TableConfig;
|
import com.walker.openocr.table.TableObject;
|
import com.walker.openocr.table.TableTextResolver;
|
import com.walker.openocr.table.config.FileConfigLoader;
|
import com.walker.openocr.vehicle.VehicleLicenseResolver;
|
import com.walker.openocr.vehicle.VehicleLicenseTable;
|
import org.junit.Test;
|
|
import java.io.File;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map;
|
|
public class TestOcrRun {
|
|
// @Test
|
public void TestRemoteAndParse(){
|
|
// 创建OCR引擎对象（初始化，无需重复创建）
|
TestOcrEngine ocrEngine = new TestOcrEngine();
|
// ocrEngine.setRemoteUrl("http://127.0.0.1:8803/ai/ocr/table");
|
ocrEngine.setRemoteUrl("http://116.198.39.83:8803/ai/ocr/table");
|
// ocrEngine.setRemoteUrl("http://101.43.168.96:8803/ai/ocr/table");
|
ocrEngine.startup();
|
|
List<IdCardConfig> idCardConfigList = new ArrayList<>(2);
|
idCardConfigList.add(new IdCardConfig());
|
|
byte[] fileContent = null;
|
String imgBase64 = null;
|
List<TextBlock> data = null;
|
Map<String, CellObject> tableData = null;
|
|
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
//~ 表格解析
|
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
// 1.请求识别服务
|
/*fileContent = FileUtils.getFileBytes(new File("D:/dev_tools/ai/demo_table_01.jpg"));
|
imgBase64 = new String(Base64.encodeBase64(fileContent));
|
data = ocrEngine.recognize("1", imgBase64); // id为业务唯一标识，存储图片拼接路径使用
|
System.out.println("返回识别结果:" + data == null ? "" : data.size());
|
|
// 2.解析表格内容
|
TableTextResolver tableTextResolver = (TableTextResolver)ocrEngine.getTextResolver(OcrType.TextTable);
|
if(tableTextResolver == null){
|
System.out.println("没有配置表格解析器:");
|
return;
|
}
|
|
// 3.获取表格数据
|
TableObject tableObject = tableTextResolver.resolve(data, ocrEngine.getTableConfigList());
|
if(tableObject == null){
|
System.out.println("没有解析到任何表格数据");
|
return;
|
}
|
|
tableData = tableObject.getTableDataMap();
|
if(tableData != null){
|
for(CellObject co : tableData.values()){
|
System.out.println(co);
|
}
|
System.out.println("----------->>>>>" + tableData.get("na_shui_ren"));
|
}*/
|
// tableObject.getRecognizeConfig().getTableTitle();
|
|
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
//~ 身份证解析
|
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
// 1.请求识别服务
|
fileContent = FileUtils.getFileBytes(new File("D:/dev_tools/ai/sfz02.jpg"));
|
imgBase64 = new String(Base64.encodeBase64(fileContent));
|
data = ocrEngine.recognize("sfz", imgBase64);
|
|
// 2.解析表格内容
|
IdCardTextResolver idCardTextResolver = (IdCardTextResolver)ocrEngine.getTextResolver(OcrType.TextIdCard);
|
if(idCardTextResolver == null){
|
System.out.println("没有配置'身份证'解析器:");
|
return;
|
}
|
|
// 3.获取身份证数据：目前仅支持正面
|
IdCardObject idCardObject = idCardTextResolver.resolve(data, idCardConfigList);
|
if(idCardObject == null){
|
System.out.println("没有解析到任何 身份证 数据");
|
return;
|
}
|
System.out.println("姓名:" + idCardObject.getName());
|
System.out.println("性别:" + idCardObject.getSex());
|
System.out.println("民族:" + idCardObject.getNation());
|
System.out.println("出生:" + idCardObject.getBirthday());
|
System.out.println("住址:" + idCardObject.getAddress());
|
System.out.println("身份证号码:" + idCardObject.getCardNo());
|
|
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
//~ 行驶证解析，2022-09-30
|
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
// FileConfigLoader vehicleConfig = new FileConfigLoader();
|
// TableConfig tableConfig = vehicleConfig.load("D:/dev_tools/workspace_2022/iplatform/iplatform-ml-openocr/src/test/resources/vehicle_license_config.txt");
|
//
|
// List<TableConfig> tableConfigs = new ArrayList<>(2);
|
// tableConfigs.add(tableConfig);
|
//
|
// // 初始化解析器(单例即可)
|
// VehicleLicenseResolver vehicleLicenseResolver = new VehicleLicenseResolver();
|
//
|
// fileContent = FileUtils.getFileBytes(new File("D:/dev_tools/ai/xsz_1.jpg"));
|
// imgBase64 = new String(Base64.encodeBase64(fileContent));
|
// data = ocrEngine.recognize("xsz001", imgBase64);
|
//
|
// VehicleLicenseTable vehicleTableObject = (VehicleLicenseTable) vehicleLicenseResolver.resolve(data, tableConfigs);
|
// tableData = vehicleTableObject.getTableDataMap();
|
// if(tableData != null){
|
// for(CellObject co : tableData.values()){
|
// System.out.println(co);
|
// }
|
// System.out.println("-----------号牌号码: " + tableData.get("hao_pai_hao_ma"));
|
// }
|
// // 识别图片中的盖章内容
|
// fileContent = FileUtils.getFileBytes(new File("D:/dev_tools/ai/红章-示例.png"));
|
// imgBase64 = new String(Base64.encodeBase64(fileContent));
|
// data = ocrEngine.recognize("10", imgBase64); // id为业务唯一标识，存储图片拼接路径使用
|
// System.out.println("返回识别结果:" + data == null ? "" : data.size());
|
}
|
}
|
