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

Revert "Using multiple instances of OCR and state recognizer for multiple threads"

Browse Source 
This reverts commit 29cb8fab21.
This commit is contained in:
Matt Hill
2014-04-04 13:58:53 -05:00
parent 3231fa47d7
commit ec8fdc8aad
2 changed files with 24 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
src/openalpr/alpr_impl.cpp
View File

@@ -26,17 +26,8 @@ AlprImpl::AlprImpl(const std::string country, const std::string runtimeDir)
{
config = new Config(country, runtimeDir);
plateDetector = new RegionDetector(config);
int numThreads = getNumThreads();
for (int i = 0; i < numThreads; i++)
{
StateIdentifier* stateIdentifier = new StateIdentifier(config);
OCR* ocr = new OCR(config);
stateIdentifiers.push_back(stateIdentifier);
ocrs.push_back(ocr);
}
stateIdentifier = new StateIdentifier(config);
ocr = new OCR(config);
setNumThreads(0);
@@ -77,28 +68,10 @@ AlprImpl::~AlprImpl()
{
delete config;
delete plateDetector;
for (int i = 0; i < stateIdentifiers.size(); i++)
{
delete stateIdentifiers[i];
delete ocrs[i];
}
stateIdentifiers.clear();
ocrs.clear();
delete stateIdentifier;
delete ocr;
}
int AlprImpl::getNumThreads()
{
// 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;
if (numThreads > tthread::thread::hardware_concurrency())
numThreads = tthread::thread::hardware_concurrency();
if (numThreads <= 0)
numThreads = 1;
return numThreads;
}
std::vector<AlprResult> AlprImpl::recognize(cv::Mat img)
{
@@ -110,11 +83,15 @@ std::vector<AlprResult> AlprImpl::recognize(cv::Mat img)
vector<Rect> plateRegions = plateDetector->detect(img);
// 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 = getNumThreads();
int numThreads = config->multithreading_cores;
if (numThreads > tthread::thread::hardware_concurrency())
numThreads = tthread::thread::hardware_concurrency();
if (numThreads <= 0)
numThreads = 1;
PlateDispatcher dispatcher(plateRegions, &img,
config, stateIdentifiers, ocrs,
config, stateIdentifier, ocr,
topN, detectRegion, defaultRegion);
// Spawn n threads to process all of the candidate regions and recognize
@@ -164,28 +141,13 @@ std::vector<AlprResult> AlprImpl::recognize(cv::Mat img)
void plateAnalysisThread(void* arg)
{
PlateDispatcher* dispatcher = (PlateDispatcher*) arg;
timespec syncstarttime;
getTime(&syncstarttime);
int threadID = dispatcher->getUniqueThreadId();
Mat img = dispatcher->getImageCopy();
if (dispatcher->config->debugGeneral)
cout << "Thread: " << tthread::this_thread::get_id() << " Initialized" << endl;
if (dispatcher->config->debugTiming)
{
timespec syncendtime;
getTime(&syncendtime);
cout << "Thread: " << tthread::this_thread::get_id() << " Synchronized initialization time in " << diffclock(syncstarttime, syncendtime) << "ms." << endl;
}
int loop_count = 0;
while (true)
{
if (dispatcher->hasPlate() == false)
break;
@@ -196,7 +158,7 @@ void plateAnalysisThread(void* arg)
// Get a single plate region from the queue
Rect plateRegion = dispatcher->nextPlate();
Mat img = dispatcher->getImageCopy();
// Parallel section
@@ -204,9 +166,6 @@ void plateAnalysisThread(void* arg)
timespec platestarttime;
getTime(&platestarttime);
StateIdentifier* stateIdentifier = dispatcher->stateIdentifiers[threadID];
OCR* ocr = dispatcher->ocrs[threadID];
LicensePlateCandidate lp(img, plateRegion, dispatcher->config);
lp.recognize();
@@ -227,7 +186,7 @@ void plateAnalysisThread(void* arg)
if (dispatcher->detectRegion)
{
char statecode[4];
plateResult.regionConfidence = stateIdentifier->recognize(img, plateRegion, statecode);
plateResult.regionConfidence = dispatcher->stateIdentifier->recognize(img, plateRegion, statecode);
if (plateResult.regionConfidence > 0)
{
plateResult.region = statecode;
@@ -235,12 +194,12 @@ void plateAnalysisThread(void* arg)
}
ocr->performOCR(lp.charSegmenter->getThresholds(), lp.charSegmenter->characters);
dispatcher->ocr->performOCR(lp.charSegmenter->getThresholds(), lp.charSegmenter->characters);
ocr->postProcessor->analyze(plateResult.region, dispatcher->topN);
dispatcher->ocr->postProcessor->analyze(plateResult.region, dispatcher->topN);
//plateResult.characters = ocr->postProcessor->bestChars;
const vector<PPResult> ppResults = ocr->postProcessor->getResults();
const vector<PPResult> ppResults = dispatcher->ocr->postProcessor->getResults();
int bestPlateIndex = 0;

30
src/openalpr/alpr_impl.h
View File

@@ -63,17 +63,15 @@ class AlprImpl
private:
int getNumThreads();
cJSON* createJsonObj(const AlprResult* result);
RegionDetector* plateDetector;
vector<StateIdentifier*> stateIdentifiers;
vector<OCR*> ocrs;
StateIdentifier* stateIdentifier;
OCR* ocr;
int topN;
bool detectRegion;
std::string defaultRegion;
cJSON* createJsonObj(const AlprResult* result);
};
@@ -83,21 +81,19 @@ class PlateDispatcher
public:
PlateDispatcher(vector<Rect> plateRegions, cv::Mat* image,
Config* config,
vector<StateIdentifier*> stateIdentifiers,
vector<OCR*> ocrs,
StateIdentifier* stateIdentifier,
OCR* ocr,
int topN, bool detectRegion, std::string defaultRegion)
{
this->plateRegions = plateRegions;
this->frame = image;
this->config = config;
this->stateIdentifiers = stateIdentifiers;
this->ocrs = ocrs;
this->stateIdentifier = stateIdentifier;
this->ocr = ocr;
this->topN = topN;
this->detectRegion = detectRegion;
this->defaultRegion = defaultRegion;
this->threadCounter = 0;
}
cv::Mat getImageCopy()
@@ -110,13 +106,6 @@ class PlateDispatcher
return img;
}
int getUniqueThreadId()
{
tthread::lock_guard<tthread::mutex> guard(mMutex);
return threadCounter++;
}
bool hasPlate()
{
bool plateAvailable;
@@ -147,8 +136,8 @@ class PlateDispatcher
}
vector<StateIdentifier*> stateIdentifiers;
vector<OCR*> ocrs;
StateIdentifier* stateIdentifier;
OCR* ocr;
Config* config;
int topN;
@@ -157,7 +146,6 @@ class PlateDispatcher
private:
int threadCounter;
tthread::mutex mMutex;
cv::Mat* frame;
vector<Rect> plateRegions;