mirror of
https://github.com/kerberos-io/openalpr-base.git
synced 2025-10-07 13:20:56 +08:00
95 lines
3.0 KiB
C++
95 lines
3.0 KiB
C++
/*
|
|
* 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/>.
|
|
*/
|
|
|
|
#ifndef CHARACTERANALYSIS_H
|
|
#define CHARACTERANALYSIS_H
|
|
|
|
#include "opencv2/imgproc/imgproc.hpp"
|
|
#include "constants.h"
|
|
#include "utility.h"
|
|
#include "config.h"
|
|
|
|
using namespace cv;
|
|
using namespace std;
|
|
|
|
class CharacterAnalysis
|
|
{
|
|
|
|
public:
|
|
CharacterAnalysis(Mat img, Config* config);
|
|
virtual ~CharacterAnalysis();
|
|
|
|
bool hasPlateMask;
|
|
Mat plateMask;
|
|
|
|
Mat bestThreshold;
|
|
vector<vector<Point> > bestContours;
|
|
vector<Vec4i> bestHierarchy;
|
|
vector<bool> bestCharSegments;
|
|
int bestCharSegmentsCount;
|
|
|
|
LineSegment topLine;
|
|
LineSegment bottomLine;
|
|
vector<Point> linePolygon;
|
|
vector<Point> charArea;
|
|
|
|
LineSegment charBoxTop;
|
|
LineSegment charBoxBottom;
|
|
LineSegment charBoxLeft;
|
|
LineSegment charBoxRight;
|
|
|
|
bool thresholdsInverted;
|
|
|
|
vector<Mat> thresholds;
|
|
vector<vector<vector<Point> > > allContours;
|
|
vector<vector<Vec4i> > allHierarchy;
|
|
vector<vector<bool> > charSegments;
|
|
|
|
void analyze();
|
|
|
|
Mat getCharacterMask();
|
|
|
|
private:
|
|
Config* config;
|
|
|
|
Mat img_gray;
|
|
|
|
Mat findOuterBoxMask( );
|
|
|
|
bool isPlateInverted();
|
|
vector<bool> filter(Mat img, vector<vector<Point> > contours, vector<Vec4i> hierarchy);
|
|
|
|
vector<bool> filterByBoxSize(vector<vector<Point> > contours, vector<bool> goodIndices, int minHeightPx, int maxHeightPx);
|
|
vector<bool> filterByParentContour( vector< vector< Point> > contours, vector<Vec4i> hierarchy, vector<bool> goodIndices);
|
|
vector<bool> filterContourHoles(vector<vector<Point> > contours, vector<Vec4i> hierarchy, vector<bool> goodIndices);
|
|
vector<bool> filterByOuterMask(vector<vector<Point> > contours, vector<Vec4i> hierarchy, vector<bool> goodIndices);
|
|
|
|
vector<Point> getCharArea();
|
|
vector<Point> getBestVotedLines(Mat img, vector<vector<Point> > contours, vector<bool> goodIndices);
|
|
//vector<Point> getCharSegmentsBetweenLines(Mat img, vector<vector<Point> > contours, vector<Point> outerPolygon);
|
|
vector<bool> filterBetweenLines(Mat img, vector<vector<Point> > contours, vector<Vec4i> hierarchy, vector<Point> outerPolygon, vector<bool> goodIndices);
|
|
|
|
bool verifySize(Mat r, float minHeightPx, float maxHeightPx);
|
|
|
|
int getGoodIndicesCount(vector<bool> goodIndices);
|
|
|
|
};
|
|
|
|
#endif // CHARACTERANALYSIS_H
|