dockerfile/openalpr-base
1
0
Fork 0
mirror of https://github.com/kerberos-io/openalpr-base.git synced 2025-10-06 19:52:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added regions of interest to ALPR interface. Currently non-functional

Browse Source
This commit is contained in:
Matt Hill
2014-08-18 23:23:03 -04:00
parent d76f122ad6
commit aac6d61773
10 changed files with 102 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
src/openalpr/alpr_impl.cpp
View File

@@ -67,10 +67,21 @@ bool AlprImpl::isLoaded()
}
AlprFullDetails AlprImpl::recognizeFullDetails(cv::Mat img)
{
std::vector<cv::Rect> regionsOfInterest;
regionsOfInterest.push_back(cv::Rect(0, 0, img.cols, img.rows));
return this->recognizeFullDetails(img, regionsOfInterest);
}
AlprFullDetails AlprImpl::recognizeFullDetails(cv::Mat img, std::vector<cv::Rect> regionsOfInterest)
{
timespec startTime;
getTime(&startTime);
if (regionsOfInterest.size() == 0)
regionsOfInterest.push_back(cv::Rect(0, 0, img.cols, img.rows));
AlprFullDetails response;
if (!img.data)
@@ -89,7 +100,7 @@ AlprFullDetails AlprImpl::recognizeFullDetails(cv::Mat img)
}
// Find all the candidate regions
response.plateRegions = plateDetector->detect(img);
response.plateRegions = plateDetector->detect(img, regionsOfInterest);
// Get the number of threads specified and make sure the value is sane (cannot be greater than CPU cores or less than 1)
int numThreads = config->multithreading_cores;
@@ -165,12 +176,27 @@ AlprFullDetails AlprImpl::recognizeFullDetails(cv::Mat img)
return response;
}
std::vector<AlprResult> AlprImpl::recognize(cv::Mat img)
std::vector<AlprResult> AlprImpl::recognize(std::string filepath, std::vector<AlprRegionOfInterest> regionsOfInterest)
{
AlprFullDetails fullDetails = recognizeFullDetails(img);
return fullDetails.results;
cv::Mat img = cv::imread(filepath, CV_LOAD_IMAGE_COLOR);
return this->recognize(img, this->convertRects(regionsOfInterest));
}
std::vector<AlprResult> AlprImpl::recognize(std::vector<unsigned char> imageBuffer, std::vector<AlprRegionOfInterest> regionsOfInterest)
{
cv::Mat img = cv::imdecode(cv::Mat(imageBuffer), 1);
return this->recognize(img, this->convertRects(regionsOfInterest));
}
std::vector<AlprResult> AlprImpl::recognize(cv::Mat img, std::vector<cv::Rect> regionsOfInterest)
{
AlprFullDetails fullDetails = recognizeFullDetails(img, regionsOfInterest);
return fullDetails.results;
}
void plateAnalysisThread(void* arg)
{
PlateDispatcher* dispatcher = (PlateDispatcher*) arg;
@@ -292,6 +318,17 @@ void plateAnalysisThread(void* arg)
cout << "Thread: " << tthread::this_thread::get_id() << " Complete" << endl;
}
std::vector<cv::Rect> AlprImpl::convertRects(std::vector<AlprRegionOfInterest> regionsOfInterest)
{
std::vector<cv::Rect> rectRegions;
for (int i = 0; i < regionsOfInterest.size(); i++)
{
rectRegions.push_back(cv::Rect(regionsOfInterest[i].x, regionsOfInterest[i].y, regionsOfInterest[i].width, regionsOfInterest[i].height));
}
return rectRegions;
}
string AlprImpl::toJson(const vector< AlprResult > results, double processing_time_ms)
{
cJSON *root, *jsonResults;