mirror of
https://github.com/kerberos-io/openalpr-base.git
synced 2025-10-06 01:27:03 +08:00
98 lines
2.6 KiB
C++
98 lines
2.6 KiB
C++
/*
|
|
* Copyright (c) 2013 New Designs Unlimited, LLC
|
|
* Opensource Automated License Plate Recognition [http://www.openalpr.com]
|
|
*
|
|
* This file is part of OpenAlpr.
|
|
*
|
|
* OpenAlpr is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License
|
|
* version 3 as published by the Free Software Foundation
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "stateidentifier.h"
|
|
|
|
|
|
StateIdentifier::StateIdentifier(Config* config)
|
|
{
|
|
this->config = config;
|
|
|
|
featureMatcher = new FeatureMatcher(config);
|
|
|
|
if (featureMatcher->isLoaded() == false)
|
|
{
|
|
cout << "Can not create detector or descriptor extractor or descriptor matcher of given types" << endl;
|
|
return;
|
|
}
|
|
|
|
featureMatcher->loadRecognitionSet(config->country);
|
|
}
|
|
|
|
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)
|
|
{
|
|
|
|
timespec startTime;
|
|
getTime(&startTime);
|
|
|
|
cvtColor(img, img, CV_BGR2GRAY);
|
|
|
|
resize(img, img, getSizeMaintainingAspect(img, 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);
|
|
vector<int> matchesArray(featureMatcher->numTrainingElements());
|
|
|
|
|
|
RecognitionResult result = featureMatcher->recognize(plateImg, true, &debugImg, true, matchesArray );
|
|
|
|
if (this->config->debugStateId)
|
|
{
|
|
|
|
|
|
displayImage(config, "State Identifier1", plateImg);
|
|
displayImage(config, "State Identifier", debugImg);
|
|
cout << result.haswinner << " : " << result.confidence << " : " << result.winner << endl;
|
|
}
|
|
|
|
|
|
if (config->debugTiming)
|
|
{
|
|
timespec endTime;
|
|
getTime(&endTime);
|
|
cout << "State Identification Time: " << diffclock(startTime, endTime) << "ms." << endl;
|
|
}
|
|
|
|
|
|
if (result.haswinner == false)
|
|
return 0;
|
|
|
|
strcpy(stateCode, result.winner.c_str());
|
|
|
|
|
|
return result.confidence;
|
|
}
|