dockerfile/openalpr-base
1
0
Fork 0
mirror of https://github.com/kerberos-io/openalpr-base.git synced 2025-10-06 07:16:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed multiline support in character analysis

Browse Source
This commit is contained in:
Matt Hill
2014-10-19 20:28:38 -04:00
parent 74b81950d5
commit 689aef8b85
1 changed files with 13 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/openalpr/segmentation/charactersegmenter.cpp
View File

@@ -84,6 +84,8 @@ CharacterSegmenter::CharacterSegmenter(PipelineData* pipeline_data)
// imgDbgGeneral.push_back(bordered); // imgDbgGeneral.push_back(bordered);
// } // }
for (uint lineidx = 0; lineidx < pipeline_data->textLines.size(); lineidx++) for (uint lineidx = 0; lineidx < pipeline_data->textLines.size(); lineidx++)
{ {
this->top = pipeline_data->textLines[lineidx].topLine; this->top = pipeline_data->textLines[lineidx].topLine;
@@ -104,7 +106,7 @@ CharacterSegmenter::CharacterSegmenter(PipelineData* pipeline_data)
vector<Mat> allHistograms; vector<Mat> allHistograms;
vector<Rect> allBoxes; vector<Rect> lineBoxes;
for (uint i = 0; i < pipeline_data->thresholds.size(); i++) for (uint i = 0; i < pipeline_data->thresholds.size(); i++)
{ {
Mat histogramMask = Mat::zeros(pipeline_data->thresholds[i].size(), CV_8U); Mat histogramMask = Mat::zeros(pipeline_data->thresholds[i].size(), CV_8U);
@@ -139,16 +141,16 @@ CharacterSegmenter::CharacterSegmenter(PipelineData* pipeline_data)
} }
for (uint z = 0; z < charBoxes.size(); z++) for (uint z = 0; z < charBoxes.size(); z++)
allBoxes.push_back(charBoxes[z]); lineBoxes.push_back(charBoxes[z]);
//drawAndWait(&histogramMask); //drawAndWait(&histogramMask);
} }
float medianCharWidth = avgCharWidth; float medianCharWidth = avgCharWidth;
vector<int> widthValues; vector<int> widthValues;
// Compute largest char width // Compute largest char width
for (uint i = 0; i < allBoxes.size(); i++) for (uint i = 0; i < lineBoxes.size(); i++)
{ {
widthValues.push_back(allBoxes[i].width); widthValues.push_back(lineBoxes[i].width);
} }
medianCharWidth = median(widthValues.data(), widthValues.size()); medianCharWidth = median(widthValues.data(), widthValues.size());
@@ -160,7 +162,7 @@ CharacterSegmenter::CharacterSegmenter(PipelineData* pipeline_data)
cout << " -- Character Segmentation Create and Score Histograms Time: " << diffclock(startTime, endTime) << "ms." << endl; cout << " -- Character Segmentation Create and Score Histograms Time: " << diffclock(startTime, endTime) << "ms." << endl;
} }
vector<Rect> candidateBoxes = getBestCharBoxes(pipeline_data->thresholds[0], allBoxes, medianCharWidth); vector<Rect> candidateBoxes = getBestCharBoxes(pipeline_data->thresholds[0], lineBoxes, medianCharWidth);
if (this->config->debugCharSegmenter) if (this->config->debugCharSegmenter)
{ {
@@ -182,18 +184,14 @@ CharacterSegmenter::CharacterSegmenter(PipelineData* pipeline_data)
getTime(&startTime); getTime(&startTime);
filterEdgeBoxes(pipeline_data->thresholds, candidateBoxes, medianCharWidth, avgCharHeight); filterEdgeBoxes(pipeline_data->thresholds, candidateBoxes, medianCharWidth, avgCharHeight);
candidateBoxes = filterMostlyEmptyBoxes(pipeline_data->thresholds, candidateBoxes); candidateBoxes = filterMostlyEmptyBoxes(pipeline_data->thresholds, candidateBoxes);
candidateBoxes = combineCloseBoxes(candidateBoxes, medianCharWidth); candidateBoxes = combineCloseBoxes(candidateBoxes, medianCharWidth);
cleanCharRegions(pipeline_data->thresholds, candidateBoxes);
cleanMostlyFullBoxes(pipeline_data->thresholds, candidateBoxes); cleanMostlyFullBoxes(pipeline_data->thresholds, candidateBoxes);
//cleanBasedOnColor(thresholds, colorFilter.colorMask, candidateBoxes);
candidateBoxes = filterMostlyEmptyBoxes(pipeline_data->thresholds, candidateBoxes); candidateBoxes = filterMostlyEmptyBoxes(pipeline_data->thresholds, candidateBoxes);
pipeline_data->charRegions = candidateBoxes;
for (uint cbox = 0; cbox < candidateBoxes.size(); cbox++)
pipeline_data->charRegions.push_back(candidateBoxes[cbox]);
if (config->debugTiming) if (config->debugTiming)
{ {
@@ -215,6 +213,9 @@ CharacterSegmenter::CharacterSegmenter(PipelineData* pipeline_data)
} }
} }
cleanCharRegions(pipeline_data->thresholds, pipeline_data->charRegions);
drawAndWait(&pipeline_data->thresholds[0]);
if (config->debugTiming) if (config->debugTiming)
{ {
timespec endTime; timespec endTime;