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

Added tagplates binary for identifying license plates (used with end-to-end benchmark)

Browse Source
This commit is contained in:
Matt Hill
2014-06-20 22:31:03 -04:00
parent 716f1ca671
commit 2c2197f1bc
2 changed files with 244 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/misc_utilities/CMakeLists.txt
View File

@@ -20,7 +20,11 @@ TARGET_LINK_LIBRARIES(classifychars
)
ADD_EXECUTABLE( benchmark benchmark.cpp )
ADD_EXECUTABLE(benchmark
benchmark/benchmark.cpp
benchmark/benchmark_utils.cpp
benchmark/endtoendtest.cpp
)
TARGET_LINK_LIBRARIES(benchmark
openalpr-static
support
@@ -37,3 +41,9 @@ TARGET_LINK_LIBRARIES(prepcharsfortraining
)
ADD_EXECUTABLE( tagplates tagplates.cpp )
TARGET_LINK_LIBRARIES(tagplates
support
${OpenCV_LIBS}
)

233
src/misc_utilities/tagplates.cpp Normal file
View File

@@ -0,0 +1,233 @@
/*
* Copyright (c) 2013 New Designs Unlimited, LLC
* Opensource Automated License Plate Recognition [http://www.openalpr.com]
*
* This file is part of OpenAlpr.
*
* OpenAlpr is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License
* version 3 as published by the Free Software Foundation
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "opencv2/highgui/highgui.hpp"
#include "opencv2/imgproc/imgproc.hpp"
#include <iostream>
#include <fstream>
#include <stdio.h>
#include <ctype.h>
#include "support/filesystem.h"
#ifdef __APPLE__
const int LEFT_ARROW_KEY = 2;
const int RIGHT_ARROW_KEY = 3;
const int SPACE_KEY = 32;
const int ENTER_KEY = 13;
const int ESCAPE_KEY = 27;
const int BACKSPACE_KEY = 8;
const int DOWN_ARROW_KEY = 1;
const int UP_ARROW_KEY= 0;
#else
const int LEFT_ARROW_KEY = 81;
const int RIGHT_ARROW_KEY = 83;
const int SPACE_KEY = 32;
const int ENTER_KEY = 10;
const int ESCAPE_KEY = 27;
const int BACKSPACE_KEY = 8;
const int DOWN_ARROW_KEY = 84;
const int UP_ARROW_KEY= 82;
#endif
using namespace std;
using namespace cv;
static bool ldragging = false;
static int xPos1 = 0;
static int yPos1 = 0;
static int xPos2 = 0;
static int yPos2 = 0;
const float ASPECT_RATIO = 2.0;
static bool rdragging = false;
static int rDragStartX = 0;
static int rDragStartY = 0;
void mouseCallback(int event, int x, int y, int flags, void* userdata)
{
if ( event == EVENT_LBUTTONDOWN )
{
ldragging = true;
xPos1 = x;
yPos1 = y;
xPos2 = x;
yPos2 = y;
}
else if ( event == EVENT_LBUTTONUP )
{
ldragging = false;
}
else if ( event == EVENT_RBUTTONDOWN )
{
rdragging = true;
rDragStartX = x;
rDragStartY = y;
}
else if ( event == EVENT_RBUTTONUP )
{
rdragging = false;
}
else if ( event == EVENT_MOUSEMOVE )
{
if (ldragging && x > xPos1 && y > yPos1)
{
int w = x - xPos1;
int h = (int) (((float) w) / ASPECT_RATIO);
xPos2 = x;
yPos2 = yPos1 + h;
}
else if (rdragging)
{
int xDiff = x - rDragStartX;
int yDiff = y - rDragStartY;
xPos1 += xDiff;
yPos1 += yDiff;
xPos2 += xDiff;
yPos2 += yDiff;
rDragStartX = x;
rDragStartY = y;
}
}
}
int main( int argc, const char** argv )
{
string country;
string inDir;
string outDir;
//Check if user specify image to process
if(argc == 4)
{
country = argv[1];
inDir = argv[2];
outDir = argv[3];
}
else
{
printf("Use:\n\t%s [country] [img input dir] [data output dir]\n",argv[0]);
printf("\tex: %s us ./usimages ./usdata\n",argv[0]);
printf("\n\n");
return 0;
}
if (DirectoryExists(inDir.c_str()) == false)
{
printf("Input dir does not exist\n");
return 0;
}
if (DirectoryExists(outDir.c_str()) == false)
{
printf("Output dir does not exist\n");
return 0;
}
vector<string> files = getFilesInDir(inDir.c_str());
vector<string> imgFiles;
sort( files.begin(), files.end(), stringCompare );
for (int i = 0; i < files.size(); i++)
{
if (hasEnding(files[i], ".png") || hasEnding(files[i], ".jpg"))
{
imgFiles.push_back(files[i]);
}
}
for (int i = 0; i< imgFiles.size(); i++)
{
cout << "Loading: " << imgFiles[i] << " (" << (i+1) << "/" << imgFiles.size() << ")" << endl;
string fullimgpath = inDir + "/" + imgFiles[i];
Mat frame = imread( fullimgpath.c_str() );
if (frame.cols == 0 || frame.rows == 0)
continue;
string curplatetag = "";
//Create a window
namedWindow("Input image", 1);
//set the callback function for any mouse event
setMouseCallback("Input image", mouseCallback, NULL);
char key = (char) cv::waitKey(50);
while (key != ENTER_KEY)
{
if ((key >= '0' && key <= '9') || (key >= 'a' && key <= 'z'))
{
curplatetag = curplatetag + (char) toupper( key );
}
else if (key == BACKSPACE_KEY)
{
curplatetag = curplatetag.substr(0, curplatetag.size() - 1);
}
Mat tmpFrame(frame.size(), frame.type());
frame.copyTo(tmpFrame);
rectangle(tmpFrame, Point(xPos1, yPos1), Point(xPos2, yPos2), Scalar(0, 0, 255), 2);
rectangle(tmpFrame, Point(xPos1, yPos1 - 35), Point(xPos1 + 175, yPos1 - 5), Scalar(255, 255, 255), CV_FILLED);
putText(tmpFrame, curplatetag, Point(xPos1 + 2, yPos1 - 10), FONT_HERSHEY_PLAIN, 1.5, Scalar(100,50,0), 2);
imshow("Input image", tmpFrame);
key = cv::waitKey(50);
}
if (curplatetag != "")
{
// Write the data to disk
ofstream outputdatafile;
std::string outputTextFile = outDir + "/" + filenameWithoutExtension(files[i]) + ".txt";
outputdatafile.open(outputTextFile.c_str());
outputdatafile << files[i] << "\t" << xPos1 << "\t" << yPos1 << "\t" << (xPos2 - xPos1) << "\t" << (yPos2 - yPos1) << "\t" << curplatetag << endl;
outputdatafile.close();
}
}
}