mirror of
https://github.com/kerberos-io/openalpr-base.git
synced 2025-10-06 18:32:45 +08:00
Added OpenALPR library
This commit is contained in:
115
src/openalpr/regiondetector.cpp
Normal file
115
src/openalpr/regiondetector.cpp
Normal file
@@ -0,0 +1,115 @@
|
||||
/*
|
||||
* 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 "regiondetector.h"
|
||||
|
||||
|
||||
RegionDetector::RegionDetector(Config* config)
|
||||
{
|
||||
this->config = config;
|
||||
// Don't scale. Can change this in the future (i.e., maximum resolution preference, or some such).
|
||||
this->scale_factor = 1.0f;
|
||||
|
||||
|
||||
// Load the cascade classifier
|
||||
if( this->plate_cascade.load( config->getCascadeRuntimeDir() + config->country + ".xml" ) )
|
||||
{
|
||||
this->loaded = true;
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("--(!)Error loading classifier\n");
|
||||
this->loaded = false;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
RegionDetector::~RegionDetector()
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
bool RegionDetector::isLoaded()
|
||||
{
|
||||
return this->loaded;
|
||||
}
|
||||
|
||||
|
||||
vector<Rect> RegionDetector::detect(Mat frame)
|
||||
{
|
||||
|
||||
Mat frame_gray;
|
||||
cvtColor( frame, frame_gray, CV_BGR2GRAY );
|
||||
|
||||
vector<Rect> regionsOfInterest = doCascade(frame_gray);
|
||||
|
||||
return regionsOfInterest;
|
||||
}
|
||||
|
||||
|
||||
/** @function detectAndDisplay */
|
||||
vector<Rect> RegionDetector::doCascade(Mat frame)
|
||||
{
|
||||
//float scale_factor = 1;
|
||||
int w = frame.size().width;
|
||||
int h = frame.size().height;
|
||||
|
||||
vector<Rect> plates;
|
||||
|
||||
equalizeHist( frame, frame );
|
||||
resize(frame, frame, Size(w * this->scale_factor, h * this->scale_factor));
|
||||
|
||||
//-- Detect plates
|
||||
timespec startTime;
|
||||
getTime(&startTime);
|
||||
|
||||
Size minSize(config->minPlateSizeWidthPx * this->scale_factor, config->minPlateSizeHeightPx * this->scale_factor);
|
||||
Size maxSize(w * config->maxPlateWidthPercent * this->scale_factor, h * config->maxPlateHeightPercent * this->scale_factor);
|
||||
|
||||
|
||||
plate_cascade.detectMultiScale( frame, plates, 1.1, 3,
|
||||
0,
|
||||
//0|CV_HAAR_SCALE_IMAGE,
|
||||
minSize, maxSize );
|
||||
|
||||
|
||||
if (config->debugTiming)
|
||||
{
|
||||
timespec endTime;
|
||||
getTime(&endTime);
|
||||
cout << "LBP Time: " << diffclock(startTime, endTime) << "ms." << endl;
|
||||
}
|
||||
|
||||
|
||||
|
||||
for( int i = 0; i < plates.size(); i++ )
|
||||
{
|
||||
plates[i].x = plates[i].x / scale_factor;
|
||||
plates[i].y = plates[i].y / scale_factor;
|
||||
plates[i].width = plates[i].width / scale_factor;
|
||||
plates[i].height = plates[i].height / scale_factor;
|
||||
}
|
||||
|
||||
return plates;
|
||||
|
||||
}
|
Reference in New Issue
Block a user