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

Updated state_identifier to use pipeline_data

Browse Source
This commit is contained in:
Matt Hill
2014-07-01 17:25:06 -04:00
parent 092e1bb3d5
commit 552caf5a05
2 changed files with 14 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/openalpr/stateidentifier.cpp
View File

@@ -43,26 +43,18 @@ StateIdentifier::~StateIdentifier()
delete featureMatcher;
}
int StateIdentifier::recognize(Mat img, Rect frame, char* stateCode)
{
Mat croppedImage = Mat(img, frame);
return this->recognize(croppedImage, stateCode);
}
// Attempts to recognize the plate. Returns a confidence level. Updates teh "stateCode" variable
// with the value of the country/state
int StateIdentifier::recognize(Mat img, char* stateCode)
// Attempts to recognize the plate. Returns a confidence level. Updates the region code and confidence
// If region is found, returns true.
bool StateIdentifier::recognize(PipelineData* pipeline_data)
{
timespec startTime;
getTime(&startTime);
cvtColor(img, img, CV_BGR2GRAY);
Mat plateImg = Mat(pipeline_data->grayImg, pipeline_data->regionOfInterest);
resize(img, img, getSizeMaintainingAspect(img, config->stateIdImageWidthPx, config->stateIdimageHeightPx));
resize(plateImg, plateImg, getSizeMaintainingAspect(plateImg, config->stateIdImageWidthPx, config->stateIdimageHeightPx));
Mat plateImg(img.size(), img.type());
//plateImg = equalizeBrightness(img);
img.copyTo(plateImg);
Mat debugImg(plateImg.size(), plateImg.type());
plateImg.copyTo(debugImg);
@@ -87,7 +79,11 @@ int StateIdentifier::recognize(Mat img, char* stateCode)
if (result.haswinner == false)
return 0;
strcpy(stateCode, result.winner.c_str());
return result.confidence;
pipeline_data->region_code = result.winner;
pipeline_data->region_confidence = result.confidence;
if (result.confidence >= 10)
return true;
return false;
}

4
src/openalpr/stateidentifier.h
View File

@@ -25,6 +25,7 @@
#include "featurematcher.h"
#include "utility.h"
#include "config.h"
#include "pipeline_data.h"
class StateIdentifier
{
@@ -33,8 +34,7 @@ class StateIdentifier
StateIdentifier(Config* config);
virtual ~StateIdentifier();
int recognize(cv::Mat img, cv::Rect frame, char* stateCode);
int recognize(cv::Mat img, char* stateCode);
bool recognize(PipelineData* pipeline_data);
//int confidence;