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

Moved polygon logic to textline

Browse Source
This commit is contained in:
Matt Hill
2014-10-13 18:19:12 -04:00
parent c156e8900e
commit 97518ecd89
4 changed files with 18 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/openalpr/textdetection/characteranalysis.cpp
View File

@@ -49,7 +49,6 @@ void CharacterAnalysis::analyze()
timespec startTime; timespec startTime;
getTime(&startTime); getTime(&startTime);
TextLine textLine;
for (uint i = 0; i < pipeline_data->thresholds.size(); i++) for (uint i = 0; i < pipeline_data->thresholds.size(); i++)
{ {
@@ -154,23 +153,18 @@ void CharacterAnalysis::analyze()
if (this->linePolygon.size() > 0) if (this->linePolygon.size() > 0)
{ {
textLine.topLine = LineSegment(this->linePolygon[0].x, this->linePolygon[0].y, this->linePolygon[1].x, this->linePolygon[1].y); LineSegment topLine = LineSegment(this->linePolygon[0].x, this->linePolygon[0].y, this->linePolygon[1].x, this->linePolygon[1].y);
textLine.bottomLine = LineSegment(this->linePolygon[3].x, this->linePolygon[3].y, this->linePolygon[2].x, this->linePolygon[2].y); LineSegment bottomLine = LineSegment(this->linePolygon[3].x, this->linePolygon[3].y, this->linePolygon[2].x, this->linePolygon[2].y);
filterBetweenLines(bestThreshold, bestContours, linePolygon); filterBetweenLines(bestThreshold, bestContours, linePolygon);
textLine.textArea = getCharArea(textLine.topLine, textLine.bottomLine); vector<Point> textArea = getCharArea(topLine, bottomLine);
if (textLine.textArea.size() > 0) TextLine textLine(textArea, topLine, bottomLine);
{
textLine.charBoxTop = LineSegment(textLine.textArea[0].x, textLine.textArea[0].y, textLine.textArea[1].x, textLine.textArea[1].y);
textLine.charBoxBottom = LineSegment(textLine.textArea[3].x, textLine.textArea[3].y, textLine.textArea[2].x, textLine.textArea[2].y);
textLine.charBoxLeft = LineSegment(textLine.textArea[3].x, textLine.textArea[3].y, textLine.textArea[0].x, textLine.textArea[0].y);
textLine.charBoxRight = LineSegment(textLine.textArea[2].x, textLine.textArea[2].y, textLine.textArea[1].x, textLine.textArea[1].y);
}
}
pipeline_data->textLines.push_back(textLine); pipeline_data->textLines.push_back(textLine);
}
this->thresholdsInverted = isPlateInverted(); this->thresholdsInverted = isPlateInverted();
} }

10
src/openalpr/textdetection/characteranalysis.h
View File

@@ -38,23 +38,13 @@ class CharacterAnalysis
cv::Mat bestThreshold; cv::Mat bestThreshold;
TextContours bestContours; TextContours bestContours;
//std::vector<std::vector<cv::Point> > bestContours;
//std::vector<cv::Vec4i> bestHierarchy;
//std::vector<bool> bestCharSegments;
//int bestCharSegmentsCount;
std::vector<cv::Point> linePolygon; std::vector<cv::Point> linePolygon;
bool thresholdsInverted; bool thresholdsInverted;
bool isTwoLine; bool isTwoLine;
std::vector<TextContours> allTextContours; std::vector<TextContours> allTextContours;
//std::vector<std::vector<std::vector<cv::Point> > > allContours;
//std::vector<std::vector<cv::Vec4i> > allHierarchy;
//std::vector<std::vector<bool> > charSegments;
void analyze(); void analyze();

12
src/openalpr/textdetection/textline.cpp
View File

@@ -20,7 +20,17 @@
#include "textline.h" #include "textline.h"
TextLine::TextLine() { TextLine::TextLine(std::vector<cv::Point> textArea, LineSegment topLine, LineSegment bottomLine) {
if (textArea.size() > 0)
{
this->textArea = textArea;
this->topLine = topLine;
this->bottomLine = bottomLine;
this->charBoxTop = LineSegment(textArea[0].x, textArea[0].y, textArea[1].x, textArea[1].y);
this->charBoxBottom = LineSegment(textArea[3].x, textArea[3].y, textArea[2].x, textArea[2].y);
this->charBoxLeft = LineSegment(textArea[3].x, textArea[3].y, textArea[0].x, textArea[0].y);
this->charBoxRight = LineSegment(textArea[2].x, textArea[2].y, textArea[1].x, textArea[1].y);
}
} }

2
src/openalpr/textdetection/textline.h
View File

@@ -25,7 +25,7 @@
class TextLine { class TextLine {
public: public:
TextLine(); TextLine(std::vector<cv::Point> textArea, LineSegment topLine, LineSegment bottomLine);
virtual ~TextLine(); virtual ~TextLine();
std::vector<cv::Point> textArea; std::vector<cv::Point> textArea;