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

Merge branch 'develop' of github.com:openalpr/openalpr into develop

Browse Source
This commit is contained in:
Matt Hill
2014-04-04 17:19:31 -05:00
parent 95c6e59610 7819aa785e
commit c982ff9715
1 changed files with 28 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
src/misc_utilities/classifychars.cpp
View File

@@ -66,20 +66,22 @@ vector<char> showCharSelection(Mat image, vector<Rect> charRegions, string state
int main( int argc, const char** argv ) int main( int argc, const char** argv )
{ {
string country;
string inDir; string inDir;
string outDir; string outDir;
Mat frame; Mat frame;
//Check if user specify image to process //Check if user specify image to process
if(argc == 3) if(argc == 4)
{ {
inDir = argv[1]; country = argv[1];
outDir = argv[2]; inDir = argv[2];
outDir = argv[3];
} }
else else
{ {
printf("Use:\n\t%s indirectory outdirectory\n",argv[0]); printf("Use:\n\t%s country indirectory outdirectory\n",argv[0]);
printf("Ex: \n\t%s ./pics/ ./out \n",argv[0]); printf("Ex: \n\t%s eu ./pics/ ./out\n",argv[0]);
return 0; return 0;
} }
@@ -92,6 +94,7 @@ int main( int argc, const char** argv )
cout << "Usage: " << endl; cout << "Usage: " << endl;
cout << "\tn -- Next plate" << endl; cout << "\tn -- Next plate" << endl;
cout << "\tp -- Previous plate" << endl; cout << "\tp -- Previous plate" << endl;
cout << "\tW -- Select image and save characters according to OCR results, then go to next image" << endl;
cout << "\ts -- Save characters" << endl; cout << "\ts -- Save characters" << endl;
cout << "\t<- and -> -- Cycle between images" << endl; cout << "\t<- and -> -- Cycle between images" << endl;
cout << "\tEnt/space -- Select plate" << endl; cout << "\tEnt/space -- Select plate" << endl;
@@ -101,8 +104,8 @@ int main( int argc, const char** argv )
cout << "\t[0-9A-Z] -- Identify a character (saves the image)" << endl; cout << "\t[0-9A-Z] -- Identify a character (saves the image)" << endl;
cout << "\tESC/Ent/Space -- Back to plate selection" << endl; cout << "\tESC/Ent/Space -- Back to plate selection" << endl;
Config* config = new Config("eu"); Config config(country);
OCR ocr(config); OCR ocr(&config);
if (DirectoryExists(inDir.c_str())) if (DirectoryExists(inDir.c_str()))
{ {
@@ -117,7 +120,7 @@ int main( int argc, const char** argv )
string fullpath = inDir + "/" + files[i]; string fullpath = inDir + "/" + files[i];
cout << fullpath << endl; cout << fullpath << endl;
frame = imread( fullpath.c_str() ); frame = imread( fullpath.c_str() );
resize(frame, frame, Size(config->ocrImageWidthPx, config->ocrImageHeightPx)); resize(frame, frame, Size(config.ocrImageWidthPx, config.ocrImageHeightPx));
imshow ("Original", frame); imshow ("Original", frame);
@@ -127,7 +130,7 @@ int main( int argc, const char** argv )
statecode[2] = '\0'; statecode[2] = '\0';
string statecodestr(statecode); string statecodestr(statecode);
CharacterRegion regionizer(frame, config); CharacterRegion regionizer(frame, &config);
if (abs(regionizer.getTopLine().angle) > 4) if (abs(regionizer.getTopLine().angle) > 4)
{ {
@@ -142,7 +145,7 @@ int main( int argc, const char** argv )
rotated.copyTo(frame); rotated.copyTo(frame);
} }
CharacterSegmenter charSegmenter(frame, regionizer.thresholdsInverted(), config); CharacterSegmenter charSegmenter(frame, regionizer.thresholdsInverted(), &config);
//ocr.cleanCharRegions(charSegmenter.thresholds, charSegmenter.characters); //ocr.cleanCharRegions(charSegmenter.thresholds, charSegmenter.characters);
@@ -202,8 +205,16 @@ int main( int argc, const char** argv )
selectedBoxes[curDashboardSelection] = !selectedBoxes[curDashboardSelection]; selectedBoxes[curDashboardSelection] = !selectedBoxes[curDashboardSelection];
showDashboard(charSegmenter.getThresholds(), selectedBoxes, curDashboardSelection); showDashboard(charSegmenter.getThresholds(), selectedBoxes, curDashboardSelection);
} }
else if (waitkey == 's' || waitkey == 'S') else if (waitkey == 's' || waitkey == 'S' || waitkey == 'W')
{ {
if (waitkey == 'W')
{
selectedBoxes[curDashboardSelection] = true;
showDashboard(charSegmenter.getThresholds(), selectedBoxes, curDashboardSelection);
const std::string& ocr_str = ocr.postProcessor->bestChars;
humanInputs.assign(ocr_str.begin(), ocr_str.end());
}
bool somethingSelected = false; bool somethingSelected = false;
bool chardataTagged = false; bool chardataTagged = false;
for (int c = 0; c < charSegmenter.getThresholds().size(); c++) for (int c = 0; c < charSegmenter.getThresholds().size(); c++)
@@ -247,6 +258,12 @@ int main( int argc, const char** argv )
cout << "Did not select any boxes" << endl; cout << "Did not select any boxes" << endl;
else if (chardataTagged == false) else if (chardataTagged == false)
cout << "You have not tagged any characters" << endl; cout << "You have not tagged any characters" << endl;
if (waitkey == 'W')
{
waitkey = 'n';
continue;
}
} }
waitkey = (char) waitKey(50); waitkey = (char) waitKey(50);