dockerfile/openalpr-base
1
0
Fork 0
mirror of https://github.com/kerberos-io/openalpr-base.git synced 2025-10-06 12:56:49 +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

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

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