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

Added detector support for regionsofinterest

Browse Source
This commit is contained in:
Matt Hill
2014-12-06 17:22:01 -05:00
parent 96b6616434
commit 6eacec7750
1 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/openalpr/detection/detectorcpu.cpp
View File

@@ -25,6 +25,7 @@ using namespace std;
namespace alpr
{
DetectorCPU::DetectorCPU(Config* config) : Detector(config) {
@@ -36,7 +37,7 @@ namespace alpr
else
{
this->loaded = false;
printf("--(!)Error loading classifier\n");
printf("--(!)Error loading CPU classifier\n");
}
}
@@ -50,12 +51,19 @@ namespace alpr
Mat frame_gray;
cvtColor( frame, frame_gray, CV_BGR2GRAY );
vector<PlateRegion> detectedRegions = doCascade(frame_gray, regionsOfInterest);
vector<PlateRegion> detectedRegions;
for (int i = 0; i < regionsOfInterest.size(); i++)
{
Mat cropped = frame_gray(regionsOfInterest[i]);
vector<PlateRegion> subRegions = doCascade(cropped, regionsOfInterest[i].x, regionsOfInterest[i].y);
for (int j = 0; j < subRegions.size(); j++)
detectedRegions.push_back(subRegions[j]);
}
return detectedRegions;
}
vector<PlateRegion> DetectorCPU::doCascade(Mat frame, std::vector<cv::Rect> regionsOfInterest)
vector<PlateRegion> DetectorCPU::doCascade(Mat frame, int offset_x, int offset_y)
{
@@ -108,8 +116,8 @@ namespace alpr
for( unsigned 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].x = (plates[i].x / scale_factor) + offset_x;
plates[i].y = (plates[i].y / scale_factor) + offset_y;
plates[i].width = plates[i].width / scale_factor;
plates[i].height = plates[i].height / scale_factor;
}