generated from zhangwei/Matrixai
823 lines
27 KiB
C++
823 lines
27 KiB
C++
#include "TrainCharacterConversionEngine.h"
|
||
|
||
|
||
using namespace ai_matrix;
|
||
|
||
namespace
|
||
{
|
||
//按照x坐标排列
|
||
bool CompareX(const SingleData &v1, const SingleData &v2)
|
||
{
|
||
return (v1.fLTX < v2.fLTX);
|
||
}
|
||
|
||
//按照y坐标排列
|
||
bool CompareY(const SingleData &v1, const SingleData &v2)
|
||
{
|
||
return (v1.fLTY < v2.fLTY);
|
||
}
|
||
|
||
//自定义比较规则
|
||
bool CmpVec(const pair<string, uint32_t> &P1, const pair<string, uint32_t> &P2)
|
||
{
|
||
return P1.second < P2.second;
|
||
}
|
||
}
|
||
|
||
TrainCharacterConversionEngine::TrainCharacterConversionEngine() {}
|
||
|
||
TrainCharacterConversionEngine::~TrainCharacterConversionEngine() {}
|
||
|
||
APP_ERROR TrainCharacterConversionEngine::Init()
|
||
{
|
||
strPort0_ = engineName_ + "_" + std::to_string(engineId_) + "_0";
|
||
strPort1_ = engineName_ + "_" + std::to_string(engineId_) + "_1";
|
||
this->modelConfig_ = Config::getins()->getModelByTrainStep2Config();
|
||
this->baseConfig_ = Config::getins()->getBaseConfig();
|
||
|
||
InitParam();
|
||
LogInfo << "TrainCharacterConversionEngine Init ok";
|
||
return APP_ERR_OK;
|
||
}
|
||
|
||
APP_ERROR TrainCharacterConversionEngine::DeInit()
|
||
{
|
||
LogInfo << "TrainCharacterConversionEngine DeInit ok";
|
||
return APP_ERR_OK;
|
||
}
|
||
|
||
/**
|
||
* 初始化参数信息
|
||
*/
|
||
void TrainCharacterConversionEngine::InitParam()
|
||
{
|
||
mapNumInfo_.clear();
|
||
iPreFrameId_ = 0;
|
||
}
|
||
|
||
/**
|
||
* 校验车型是否符合验证
|
||
* inParam : int classId 大框类别id
|
||
* : const std::string &trainNum 车型字符信息
|
||
* outParam: N/A
|
||
* return : true:符合; false:不符合
|
||
*/
|
||
bool TrainCharacterConversionEngine::authTransNum(int classId, const std::string &trainNum)
|
||
{
|
||
switch (classId)
|
||
{
|
||
case TRAIN_HEAD: // 车头上的编号
|
||
break;
|
||
case K_TRAIN_NUM: // 编号 矿车、煤炭漏斗车（兖矿自备、枣矿自备）
|
||
{
|
||
if (trainNum == "K13" || trainNum == "KM100AH")
|
||
{
|
||
return true;
|
||
}
|
||
std::regex reg;
|
||
switch (trainNum.size())
|
||
{
|
||
case 4:
|
||
reg = "^K[1FMZ]{1}[36789]{1}[018ABKNT]{1}";
|
||
break;
|
||
case 5:
|
||
reg = "^K[1M]{1}[368]{1}[1ABDN]{1}[AGKNT]{1}";
|
||
break;
|
||
case 6:
|
||
reg = "^K[MF]{1}[69]{1}[80]{1}A[KH]{1}";
|
||
break;
|
||
default:
|
||
// LogWarn << "Unknow train，classId:" << classId << " trainNum:" << trainNum;
|
||
return false;
|
||
}
|
||
return std::regex_match(trainNum, reg);
|
||
}
|
||
case C_TRAIN_NUM: // 敞车 特殊车型 C5D CF CFK
|
||
{
|
||
if (trainNum == "CF" || trainNum == "C62A(N)")
|
||
{
|
||
return true;
|
||
}
|
||
std::regex reg;
|
||
switch (trainNum.size())
|
||
{
|
||
case 3:
|
||
reg = "^C[15678F]{1}[0123456DK]{1}";
|
||
break;
|
||
case 4:
|
||
reg = "^C[1678]{1}[012346]{1}[0ABCEFHKMTY]{1}";
|
||
break;
|
||
case 5:
|
||
reg = "^C[1678]{1}[01246]{1}[0ABCEY]{1}[KAFHNT]{1}";
|
||
break;
|
||
case 6:
|
||
reg = "^C[17]{1}0[0E]{1}[A-]{1}[AH]{1}";
|
||
break;
|
||
default:
|
||
// LogWarn << "Unknow train，classId:" << classId << " trainNum:" << trainNum;
|
||
return false;
|
||
}
|
||
return std::regex_match(trainNum, reg);
|
||
}
|
||
case P_TRAIN_NUM: // 编号 棚车
|
||
{
|
||
if (trainNum == "P70(H)" || trainNum == "P62(N)")
|
||
{
|
||
return true;
|
||
}
|
||
std::regex reg;
|
||
switch (trainNum.size())
|
||
{
|
||
case 3:
|
||
reg = "^P[1678]{1}[012345]{1}";
|
||
break;
|
||
case 4:
|
||
reg = "^P[678]{1}[023456]{1}[ABHKNST]{1}";
|
||
break;
|
||
case 5:
|
||
reg = "^P6[24]{1}[ANG]{1}[KT]{1}";
|
||
break;
|
||
default:
|
||
// LogWarn << "Unknow train，classId:" << classId << " trainNum:" << trainNum;
|
||
return false;
|
||
}
|
||
return std::regex_match(trainNum, reg);
|
||
}
|
||
case G_TRAIN_NUM: // 编号 罐车
|
||
{
|
||
if (trainNum == "GY100SK")
|
||
{
|
||
return true;
|
||
}
|
||
std::regex reg;
|
||
switch (trainNum.size())
|
||
{
|
||
case 2:
|
||
reg = "^G[HLS]{1}";
|
||
break;
|
||
case 3:
|
||
reg = "^[GL]{1}[1567HL]{1}[01246789K]{1}";
|
||
break;
|
||
case 4:
|
||
reg = "^[GU]{1}[167FJLNQSWY]{1}[170689]{1}[KSABDGTM075W]{1}";
|
||
break;
|
||
case 5:
|
||
reg = "^[GU]{1}[17FHNQY6]{1}[1706A89]{1}[SBDAM057W]{1}[KEHTB0ZAS]{1}";
|
||
break;
|
||
case 6:
|
||
reg = "^[UG]{1}[6YH]{1}[1489A]{1}[057W]{1}[Z0LSA]{1}[KSAT]{1}";
|
||
break;
|
||
default:
|
||
// LogWarn << "Unknow train，classId:" << classId << " trainNum:" << trainNum;
|
||
return false;
|
||
}
|
||
return std::regex_match(trainNum, reg);
|
||
}
|
||
case NX_TRAIN_NUM: // 编号 平车
|
||
{
|
||
if (trainNum == "N6")
|
||
{
|
||
return true;
|
||
}
|
||
std::regex reg;
|
||
switch (trainNum.size())
|
||
{
|
||
case 3:
|
||
reg = "^[NQX]{1}[1234678T]{1}[0124567HK]{1}";
|
||
break;
|
||
case 4:
|
||
reg = "^[BNX]{1}[16X]{1}[17BC]{1}[07AGKT]{1}";
|
||
break;
|
||
case 5:
|
||
reg = "^[NX]{1}[16NX]{1}[17C]{1}[07AGT]{1}[KTABH]{1}";
|
||
break;
|
||
case 6:
|
||
reg = "^NX[170]{2}[AB]{1}[KTHF]{1}";
|
||
break;
|
||
case 7:
|
||
reg = "^NX70[AB]{1}[-]{1}[KTHF]{1}";
|
||
break;
|
||
default:
|
||
// LogWarn << "Unknow train，classId:" << classId << " trainNum:" << trainNum;
|
||
return false;
|
||
}
|
||
return std::regex_match(trainNum, reg);
|
||
}
|
||
case J_TRAIN_NUM: // 编号 牲畜车 特种车
|
||
{
|
||
if (trainNum == "TP64GK")
|
||
{
|
||
return true;
|
||
}
|
||
std::regex reg;
|
||
switch (trainNum.size())
|
||
{
|
||
case 2:
|
||
reg = "^[DT]{1}[2678]{1}";
|
||
break;
|
||
case 3:
|
||
reg = "^[DST]{1}[12346789LQ]{1}[012578ADFGP]{1}";
|
||
break;
|
||
case 4:
|
||
reg = "^[DJNT]{1}[12356AFKQS]{1}[012345678DFQS]{1}[A12345679GHKQ]{1}";
|
||
break;
|
||
case 5:
|
||
reg = "^[DJQT]{1}[2KNSH1P]{1}[613XQ2]{1}[A761234B]{1}[KA70G]{1}";
|
||
break;
|
||
default:
|
||
// LogWarn << "Unknow train， classId: " << classId << " trainNum:" << trainNum;
|
||
return false;
|
||
}
|
||
return std::regex_match(trainNum, reg);
|
||
}
|
||
default:
|
||
LogWarn << "Unknow train，classId:" << classId << " trainNum:" << trainNum;
|
||
return false;
|
||
}
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* 过滤第二步字段代号的误识别
|
||
* inParam : std::vector<SingleData> &vecObjs
|
||
* : TargetMaxLen iMaxLen
|
||
* outParam: N/A
|
||
* return : N/A
|
||
*/
|
||
void TrainCharacterConversionEngine::filterSingleData(std::vector<SingleData> &vecObjs, TargetMaxLen iMaxLen)
|
||
{
|
||
/*误识别场景
|
||
例： 车厢容积属性只有2位时，但因误识别导致自重的某个字符字段代号错误，划归到容积中，这样实际只有2位的容积变为3位。导致选优时错误。
|
||
只处理1个字段代号的误识别，若有多个字段代号同时错误，无法区分哪些字符字段代号是正确的。(字段代号误识别较多时，则增强模型训练)
|
||
|
||
排除一个最大和最小值后计算平局坐标值，每个框左上角和平局值左上角高度差大于平局值的高度则剔除.
|
||
*/
|
||
if (vecObjs.size() != iMaxLen || vecObjs.size() <= 2)
|
||
{
|
||
return;
|
||
}
|
||
|
||
std::deque<SingleData> dqObjs(vecObjs.begin(), vecObjs.end());
|
||
std::sort(dqObjs.begin(), dqObjs.end(), [](SingleData &a, SingleData &b)
|
||
{
|
||
return a.fLTY < b.fLTY;
|
||
});
|
||
dqObjs.pop_front();
|
||
dqObjs.pop_back();
|
||
|
||
SingleData singleDataAvg;
|
||
for (auto & dqObj : dqObjs)
|
||
{
|
||
singleDataAvg.fLTX += dqObj.fLTX;
|
||
singleDataAvg.fLTY += dqObj.fLTY;
|
||
singleDataAvg.fRBX += dqObj.fRBX;
|
||
singleDataAvg.fRBY += dqObj.fRBY;
|
||
}
|
||
singleDataAvg.fLTX = singleDataAvg.fLTX / dqObjs.size();
|
||
singleDataAvg.fLTY = singleDataAvg.fLTY / dqObjs.size();
|
||
singleDataAvg.fRBX = singleDataAvg.fRBX / dqObjs.size();
|
||
singleDataAvg.fRBY = singleDataAvg.fRBY / dqObjs.size();
|
||
float fAvgHeight = singleDataAvg.fRBY - singleDataAvg.fLTY;
|
||
|
||
//剔除位置错误的字符信息
|
||
for (auto iter = vecObjs.begin(); iter != vecObjs.end(); iter++)
|
||
{
|
||
float fTemp = fabs(iter->fLTY - singleDataAvg.fLTY);
|
||
if (fTemp > (fAvgHeight - 5))
|
||
{
|
||
std::string strOne = this->modelConfig_.vecClass.at(iter->iClassId);
|
||
LogWarn << "engineId:" << engineId_ << " " << strOne << " Line wrong ";
|
||
vecObjs.erase(iter);
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 属性框内容转换
|
||
* inParam : std::map<int, std::vector<SingleData>> &mapLine
|
||
* outParam: TransSubData &transSubData
|
||
* return : N/A
|
||
*/
|
||
void TrainCharacterConversionEngine::transPro(Step2ResultData &step2ResultData, std::map<int, std::vector<SingleData>> &mapLine)
|
||
{
|
||
//载重
|
||
if (mapLine.find(0) != mapLine.end())
|
||
{
|
||
TransInfo loadInfo;
|
||
loadInfo.iLine = 0;
|
||
std::string strTemp;
|
||
for (auto j = 0; j < mapLine.at(0).size(); j++)
|
||
{
|
||
//过滤非数字
|
||
std::string strOne = this->modelConfig_.vecClass.at(mapLine.at(0).at(j).iClassId);
|
||
if (strOne[0] < '0' || strOne[0] > '9')
|
||
{
|
||
LogWarn << "engineId:" << engineId_ << " " << strOne << " not digit in load";
|
||
continue;
|
||
}
|
||
strTemp += strOne;
|
||
loadInfo.vecValue.emplace_back(strOne);
|
||
loadInfo.vecScore.emplace_back(mapLine.at(0).at(j).fScore);
|
||
step2ResultData.fSubScoreSum += mapLine.at(0).at(j).fScore;
|
||
}
|
||
|
||
//匹配正则表达式
|
||
//std::regex reg("^[0-9]{2}");
|
||
std::regex reg("^([4,5,6,7,8,9]{1}[0-9]{1}|[1]{1}[0-3]{1}[0-9]{1})$"); //(第1位：棚车:4 5; K自备车:1 5 7 8 9; 平车:7; 罐车:6 7; 敞车:6 7 8)
|
||
if (std::regex_match(strTemp, reg))
|
||
{
|
||
loadInfo.IsChkFlag = true;
|
||
}
|
||
loadInfo.strTmpResult = strTemp;
|
||
step2ResultData.vecTransInfo.emplace_back(loadInfo);
|
||
}
|
||
|
||
//自重
|
||
if (mapLine.find(1) != mapLine.end())
|
||
{
|
||
this->filterSingleData(mapLine.at(1), SELF_MAXLEN);
|
||
TransInfo selfInfo;
|
||
selfInfo.iLine = 1;
|
||
std::string strTemp;
|
||
for (auto j = 0; j < mapLine.at(1).size(); j++)
|
||
{
|
||
//过滤非数字
|
||
std::string strOne = this->modelConfig_.vecClass.at(mapLine.at(1).at(j).iClassId);
|
||
if (strOne[0] < '0' || strOne[0] > '9')
|
||
{
|
||
LogWarn << "engineId:" << engineId_ << " " << strOne << " not digit in self";
|
||
continue;
|
||
}
|
||
strTemp += strOne;
|
||
selfInfo.vecValue.emplace_back(strOne);
|
||
selfInfo.vecScore.emplace_back(mapLine.at(1).at(j).fScore);
|
||
step2ResultData.fSubScoreSum += mapLine.at(1).at(j).fScore;
|
||
}
|
||
|
||
//匹配正则表达式
|
||
std::regex reg("^[1,2,3]{1}[0-9]{1,2}$");
|
||
if (std::regex_match(strTemp, reg))
|
||
{
|
||
selfInfo.IsChkFlag = true;
|
||
}
|
||
selfInfo.strTmpResult = strTemp;
|
||
step2ResultData.vecTransInfo.emplace_back(selfInfo);
|
||
}
|
||
|
||
//容积
|
||
if (mapLine.find(2) != mapLine.end())
|
||
{
|
||
this->filterSingleData(mapLine.at(2), VOLUME_MAXLEN);
|
||
TransInfo volumeInfo;
|
||
volumeInfo.iLine = 2;
|
||
std::string strTemp;
|
||
for (auto j = 0; j < mapLine.at(2).size(); j++)
|
||
{
|
||
//过滤非数字
|
||
std::string strOne = this->modelConfig_.vecClass.at(mapLine.at(2).at(j).iClassId);
|
||
if (strOne[0] < '0' || strOne[0] > '9')
|
||
{
|
||
LogWarn << "engineId:" << engineId_ << " " << strOne << " not digit in volume";
|
||
continue;
|
||
}
|
||
strTemp += strOne;
|
||
volumeInfo.vecValue.emplace_back(strOne);
|
||
volumeInfo.vecScore.emplace_back(mapLine.at(2).at(j).fScore);
|
||
step2ResultData.fSubScoreSum += mapLine.at(2).at(j).fScore;
|
||
}
|
||
|
||
//匹配正则表达式
|
||
//std::regex reg("^[0-9]{2,3}$");
|
||
std::regex reg("^[1,6,7,8,9]{1}[0-9]{1,2}$"); //(第1位：棚车:1; K自备车:1或9; 罐车:6或7; 敞车:7或8)
|
||
if (std::regex_match(strTemp, reg))
|
||
{
|
||
volumeInfo.IsChkFlag = true;
|
||
}
|
||
|
||
volumeInfo.strTmpResult = strTemp;
|
||
step2ResultData.vecTransInfo.emplace_back(volumeInfo);
|
||
}
|
||
|
||
//换长
|
||
if (mapLine.find(3) != mapLine.end())
|
||
{
|
||
TransInfo changeInfo;
|
||
changeInfo.iLine = 3;
|
||
std::string strTemp;
|
||
for (auto j = 0; j < mapLine.at(3).size(); j++)
|
||
{
|
||
//过滤非数字
|
||
std::string strOne = this->modelConfig_.vecClass.at(mapLine.at(3).at(j).iClassId);
|
||
if (strOne[0] < '0' || strOne[0] > '9')
|
||
{
|
||
LogWarn << "engineId:" << engineId_ << " " << strOne << " not digit in change";
|
||
continue;
|
||
}
|
||
strTemp += strOne;
|
||
changeInfo.vecValue.emplace_back(strOne);
|
||
changeInfo.vecScore.emplace_back(mapLine.at(3).at(j).fScore);
|
||
step2ResultData.fSubScoreSum += mapLine.at(3).at(j).fScore;
|
||
}
|
||
|
||
//匹配正则表达式
|
||
//std::regex reg("^[0-9]{2}$");
|
||
std::regex reg("^[1]{1}[0-9]{1}$");
|
||
if (std::regex_match(strTemp, reg))
|
||
{
|
||
changeInfo.IsChkFlag = true;
|
||
}
|
||
|
||
changeInfo.strTmpResult = strTemp;
|
||
step2ResultData.vecTransInfo.emplace_back(changeInfo);
|
||
}
|
||
|
||
//罐车容量计表
|
||
if (mapLine.find(4) != mapLine.end())
|
||
{
|
||
TransInfo volumeSurfaceInfo;
|
||
volumeSurfaceInfo.iLine = 4;
|
||
std::string strTemp;
|
||
for (auto j = 0; j < mapLine.at(4).size(); j++)
|
||
{
|
||
strTemp += this->modelConfig_.vecClass.at(mapLine.at(4).at(j).iClassId);
|
||
volumeSurfaceInfo.vecValue.emplace_back(this->modelConfig_.vecClass.at(mapLine.at(4).at(j).iClassId));
|
||
volumeSurfaceInfo.vecScore.emplace_back(mapLine.at(4).at(j).fScore);
|
||
step2ResultData.fSubScoreSum += mapLine.at(4).at(j).fScore;
|
||
}
|
||
|
||
volumeSurfaceInfo.IsChkFlag = true; //暂不校验
|
||
step2ResultData.vecTransInfo.emplace_back(volumeSurfaceInfo);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 车号框内容转换
|
||
* inParam : std::map<int, std::vector<SingleData>> &mapLine
|
||
* outParam: TransSubData &transSubData
|
||
* return : N/A
|
||
*/
|
||
void TrainCharacterConversionEngine::transNum(Step2ResultData &step2ResultData, std::map<int, std::vector<SingleData>> &mapLine)
|
||
{
|
||
//车型
|
||
if (mapLine.find(0) != mapLine.end())
|
||
{
|
||
TransInfo typeInfo;
|
||
typeInfo.iLine = 0;
|
||
std::string strTemp;
|
||
for (auto j = 0; j < mapLine.at(0).size(); j++)
|
||
{
|
||
std::string strOne = this->modelConfig_.vecClass.at(mapLine.at(0).at(j).iClassId);
|
||
|
||
strTemp += strOne;
|
||
typeInfo.vecValue.emplace_back(strOne);
|
||
typeInfo.vecScore.emplace_back(mapLine.at(0).at(j).fScore);
|
||
step2ResultData.fSubScoreSum += mapLine.at(0).at(j).fScore;
|
||
}
|
||
|
||
//校验车型是否符合验证
|
||
typeInfo.IsChkFlag = this->authTransNum(step2ResultData.iClassId, strTemp);
|
||
LogDebug << "--->>> 符合正则吗？" << typeInfo.IsChkFlag << " --- " << strTemp;
|
||
|
||
typeInfo.strTmpResult = strTemp;
|
||
step2ResultData.vecTransInfo.emplace_back(typeInfo);
|
||
}
|
||
|
||
// 编号
|
||
if (mapLine.find(1) != mapLine.end())
|
||
{
|
||
TransInfo numInfo;
|
||
numInfo.iLine = 1;
|
||
std::string strTemp;
|
||
for (auto j = 0; j < mapLine.at(1).size(); j++)
|
||
{
|
||
std::string strOne = this->modelConfig_.vecClass.at(mapLine.at(1).at(j).iClassId);
|
||
// 过滤非数字
|
||
if (strOne[0] < '0' || strOne[0] > '9')
|
||
{
|
||
LogWarn << "engineId:" << engineId_ << " " << strOne << " not digit in num";
|
||
continue;
|
||
}
|
||
strTemp += strOne;
|
||
numInfo.vecValue.emplace_back(strOne);
|
||
numInfo.vecScore.emplace_back(mapLine.at(1).at(j).fScore);
|
||
step2ResultData.fSubScoreSum += mapLine.at(1).at(j).fScore;
|
||
}
|
||
|
||
//匹配正则表达式
|
||
std::regex reg("^[0-9]{5,7}$"); //车厢编号，固定7位
|
||
if (std::regex_match(strTemp, reg))
|
||
{
|
||
numInfo.IsChkFlag = true;
|
||
}
|
||
numInfo.strTmpResult = strTemp;
|
||
|
||
// LogDebug << "--->>> 符合正则吗？" << numInfo.IsChkFlag << " --- " << strTemp;
|
||
step2ResultData.vecTransInfo.emplace_back(numInfo);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 车头框内容转换
|
||
* inParam : std::map<int, std::vector<SingleData>> &mapLine
|
||
* outParam: TransSubData &transSubData
|
||
* return : N/A
|
||
*/
|
||
void TrainCharacterConversionEngine::transHead(Step2ResultData &step2ResultData, std::map<int, std::vector<SingleData>> &mapLine)
|
||
{
|
||
//车头型号
|
||
if (mapLine.find(0) != mapLine.end())
|
||
{
|
||
TransInfo typeInfo;
|
||
typeInfo.iLine = 0;
|
||
std::string strTemp;
|
||
for (auto j = 0; j < mapLine.at(0).size(); j++)
|
||
{
|
||
strTemp += this->modelConfig_.vecClass.at(mapLine.at(0).at(j).iClassId);
|
||
typeInfo.vecValue.emplace_back(this->modelConfig_.vecClass.at(mapLine.at(0).at(j).iClassId));
|
||
typeInfo.vecScore.emplace_back(mapLine.at(0).at(j).fScore);
|
||
step2ResultData.fSubScoreSum += mapLine.at(0).at(j).fScore;
|
||
}
|
||
|
||
//不做结果校验，IsChkFlag默认为true
|
||
typeInfo.IsChkFlag = true;
|
||
step2ResultData.vecTransInfo.emplace_back(typeInfo);
|
||
}
|
||
|
||
//车头编号
|
||
if (mapLine.find(1) != mapLine.end())
|
||
{
|
||
TransInfo numInfo;
|
||
numInfo.iLine = 1;
|
||
std::string strTemp;
|
||
for (auto j = 0; j < mapLine.at(1).size(); j++)
|
||
{
|
||
std::string strOne = this->modelConfig_.vecClass.at(mapLine.at(1).at(j).iClassId);
|
||
//过滤非数字
|
||
if (strOne[0] < '0' || strOne[0] > '9')
|
||
{
|
||
LogWarn << "engineId:" << engineId_ << " " << strOne << " not digit in headnum";
|
||
continue;
|
||
}
|
||
strTemp += strOne;
|
||
numInfo.vecValue.emplace_back(strOne);
|
||
numInfo.vecScore.emplace_back(mapLine.at(1).at(j).fScore);
|
||
step2ResultData.fSubScoreSum += mapLine.at(1).at(j).fScore;
|
||
}
|
||
|
||
//不做结果校验，IsChkFlag默认为true
|
||
numInfo.IsChkFlag = true;
|
||
step2ResultData.vecTransInfo.emplace_back(numInfo);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 验证集装箱号是否满足规则(前10位依次与2^0 ~2^9相乘）/11 所得余数，即是校验位)
|
||
* inParam : std::string &strContainerNo 集装箱号信息
|
||
* outParam: N/A
|
||
* return : true(校验通过)/false(校验失败)
|
||
*/
|
||
bool TrainCharacterConversionEngine::verifyContainerNo(std::string &strContainerNo)
|
||
{
|
||
bool bChkFlag = false;
|
||
|
||
if(strContainerNo.length() != 11)
|
||
{
|
||
return bChkFlag;
|
||
}
|
||
|
||
int iSum = 0;
|
||
for (int i = 0; i < strContainerNo.length()-1; ++i)
|
||
{
|
||
iSum += this->mapExchange_[strContainerNo.substr(i, 1)] * int(pow(2.0, i));
|
||
}
|
||
|
||
//当校验位等于10时要继续模运算,iSum % 11 % 10,保证最终结果为0~9之间的数
|
||
int iChkValue = iSum % 11 % 10;
|
||
if (iChkValue == atoi(strContainerNo.substr(strContainerNo.length()-1, 1).c_str()))
|
||
{
|
||
bChkFlag = true;
|
||
}
|
||
return bChkFlag;
|
||
}
|
||
|
||
/**
|
||
* 集装箱框内容转换
|
||
* inParam : std::map<int, std::vector<SingleData>> &mapLine
|
||
* outParam: TransSubData &transSubData
|
||
* return : N/A
|
||
*/
|
||
void TrainCharacterConversionEngine::transContainerNum(Step2ResultData &step2ResultData, std::map<int, std::vector<SingleData>> &mapLine)
|
||
{
|
||
TransInfo info;
|
||
info.iLine = 0;
|
||
// 箱主代码+设备识别码（4位英文字母）
|
||
if (mapLine.find(0) != mapLine.end())
|
||
{
|
||
std::string strTemp;
|
||
for (auto j = 0; j < mapLine.at(0).size(); j++)
|
||
{
|
||
std::string strOne = this->modelConfig_.vecClass.at(mapLine.at(0).at(j).iClassId);
|
||
// 过滤非字母
|
||
if (strOne[0] < 'A' || strOne[0] > 'Z')
|
||
{
|
||
// LogDebug << strOne << " not A-Z in container";
|
||
continue;
|
||
}
|
||
strTemp += strOne;
|
||
info.vecValue.emplace_back(strOne);
|
||
info.vecScore.emplace_back(mapLine.at(0).at(j).fScore);
|
||
step2ResultData.fSubScoreSum += mapLine.at(0).at(j).fScore;
|
||
}
|
||
info.strTmpResult += strTemp;
|
||
}
|
||
|
||
// 箱号（6位数字）
|
||
if (mapLine.find(1) != mapLine.end())
|
||
{
|
||
std::string strTemp;
|
||
for (auto j = 0; j < mapLine.at(1).size(); j++)
|
||
{
|
||
std::string strOne = this->modelConfig_.vecClass.at(mapLine.at(1).at(j).iClassId);
|
||
// 过滤非数字
|
||
if (strOne[0] < '0' || strOne[0] > '9')
|
||
{
|
||
// LogDebug << "engineId:" << engineId_ << " " << strOne << " not digit in num";
|
||
continue;
|
||
}
|
||
strTemp += strOne;
|
||
info.vecValue.emplace_back(strOne);
|
||
info.vecScore.emplace_back(mapLine.at(1).at(j).fScore);
|
||
step2ResultData.fSubScoreSum += mapLine.at(1).at(j).fScore;
|
||
}
|
||
info.strTmpResult += strTemp;
|
||
}
|
||
|
||
//校验码（1位数字）
|
||
if (mapLine.find(2) != mapLine.end())
|
||
{
|
||
std::string strTemp;
|
||
for (auto j = 0; j < mapLine.at(2).size(); j++)
|
||
{
|
||
//过滤非数字
|
||
std::string strOne = this->modelConfig_.vecClass.at(mapLine.at(2).at(j).iClassId);
|
||
if (strOne[0] < '0' || strOne[0] > '9')
|
||
{
|
||
// LogDebug << "engineId:" << engineId_ << " " << strOne << " not digit in container_2";
|
||
continue;
|
||
}
|
||
strTemp += strOne;
|
||
info.vecValue.emplace_back(strOne);
|
||
info.vecScore.emplace_back(mapLine.at(2).at(j).fScore);
|
||
step2ResultData.fSubScoreSum += mapLine.at(2).at(j).fScore;
|
||
}
|
||
info.strTmpResult += strTemp;
|
||
}
|
||
|
||
// LogDebug << " -- " << info.strTmpResult;
|
||
|
||
if (this->verifyContainerNo(info.strTmpResult))
|
||
{
|
||
info.IsChkFlag = true;
|
||
}
|
||
step2ResultData.vecTransInfo.emplace_back(info);
|
||
}
|
||
|
||
|
||
/**
|
||
* 记录在画面中间位置识别的车号
|
||
* inParam : const TransSubData &transSubData 车号信息
|
||
: const std::shared_ptr<ProcessData> &pProcessData 帧数据
|
||
* outParam: N/A
|
||
* return : N/A
|
||
*/
|
||
void TrainCharacterConversionEngine::recordNum(const Step2ResultData &step2ResultData, const int iFrameId)
|
||
{
|
||
float iCenterX = step2ResultData.fLTX + (step2ResultData.fRBX - step2ResultData.fLTX) / 2;
|
||
float iImgL = IMAGE_WIDTH / 3;
|
||
float iImgR = IMAGE_WIDTH / 3 * 2;
|
||
if (iCenterX <= iImgL || iCenterX >= iImgR)
|
||
{
|
||
return;
|
||
}
|
||
|
||
std::string strValue;
|
||
for (const auto& transInfo : step2ResultData.vecTransInfo)
|
||
{
|
||
if (transInfo.iLine != 1)
|
||
{
|
||
continue;
|
||
}
|
||
for (const auto & j : transInfo.vecValue)
|
||
{
|
||
strValue += j;
|
||
}
|
||
}
|
||
|
||
bool bIntervalFlag = ((int)(iFrameId - iPreFrameId_) > iSkipInterval_ * 2);
|
||
if (mapNumInfo_.find(strValue) == mapNumInfo_.end())
|
||
{
|
||
// LogDebug << " frameid:" << iFrameId << " NUM:" << strValue
|
||
// << " bIntervalFlag:" << bIntervalFlag;
|
||
if (bIntervalFlag)
|
||
{
|
||
mapNumInfo_[strValue] = iFrameId;
|
||
}
|
||
}
|
||
iPreFrameId_ = iFrameId;
|
||
}
|
||
|
||
APP_ERROR TrainCharacterConversionEngine::Process()
|
||
{
|
||
int iRet = APP_ERR_OK;
|
||
while (!isStop_)
|
||
{
|
||
std::shared_ptr<void> pVoidData0 = nullptr;
|
||
inputQueMap_[strPort0_]->pop(pVoidData0);
|
||
if (nullptr == pVoidData0)
|
||
{
|
||
usleep(1000); //1ms
|
||
continue;
|
||
}
|
||
|
||
std::shared_ptr<VStep2OutputData> pVStep2OutputData = std::static_pointer_cast<VStep2OutputData>(pVoidData0);
|
||
|
||
for (auto & i : pVStep2OutputData->vecStep2ResultData)
|
||
{
|
||
std::vector<SingleData> vecSingleData = i.vecSingleData;
|
||
|
||
std::map<int, std::vector<SingleData>> mapLine;
|
||
for (auto & j : vecSingleData)
|
||
{
|
||
mapLine[j.iLine].push_back(j);
|
||
}
|
||
float iWidth = i.fRBX - i.fLTX;
|
||
float iHeight = i.fRBY - i.fLTY;
|
||
bool bSortByX = iHeight < (iWidth * 1.5); // 竖排集装箱按y坐标排序
|
||
//每一行按x坐标排序
|
||
for (auto & it : mapLine)
|
||
{
|
||
if (it.second.size() > 2)
|
||
{
|
||
std::set<float> setX;
|
||
std::set<float> setY;
|
||
for (auto & j : it.second)
|
||
{
|
||
setX.insert(j.fLTX);
|
||
setY.insert(j.fLTY);
|
||
}
|
||
float X = *setX.rbegin() - *setX.begin();
|
||
float Y = *setY.rbegin() - *setY.begin();
|
||
bSortByX = (X > Y);
|
||
}
|
||
|
||
if (bSortByX)
|
||
{
|
||
std::sort(it.second.begin(), it.second.end(), CompareX);
|
||
}
|
||
else
|
||
{
|
||
std::sort(it.second.begin(), it.second.end(), CompareY);
|
||
}
|
||
|
||
std::string strTemp;
|
||
for (auto & j : it.second)
|
||
{
|
||
strTemp += this->modelConfig_.vecClass.at(j.iClassId);
|
||
}
|
||
}
|
||
|
||
switch (i.iClassId)
|
||
{
|
||
case TRAIN_HEAD:
|
||
this->transHead(i, mapLine);
|
||
// this->recordNum(pVStep2OutputData->vecStep2ResultData[i], pVStep2OutputData->iFrameId);
|
||
break;
|
||
case TRAIN_PRO:
|
||
this->transPro(i, mapLine);
|
||
break;
|
||
case K_TRAIN_NUM ... NX_TRAIN_NUM:
|
||
this->transNum(i, mapLine);
|
||
// this->recordNum(pVStep2OutputData->vecStep2ResultData[i], pVStep2OutputData->iFrameId);
|
||
break;
|
||
case CONTAINERNUM:
|
||
this->transContainerNum(i, mapLine);
|
||
break;
|
||
default:
|
||
break;
|
||
}
|
||
}
|
||
|
||
outputQueMap_[strPort0_]->push(std::static_pointer_cast<void>(pVStep2OutputData), true);
|
||
outputQueMap_[strPort1_]->push(std::static_pointer_cast<void>(pVStep2OutputData), true);
|
||
|
||
if (pVStep2OutputData->bIsEnd)
|
||
{
|
||
//初始化参数变量
|
||
InitParam();
|
||
}
|
||
}
|
||
|
||
return APP_ERR_OK;
|
||
}
|