mirror of
https://github.com/kerberos-io/openalpr-base.git
synced 2025-10-06 07:37:00 +08:00
131 lines
3.6 KiB
C++
131 lines
3.6 KiB
C++
/*
|
|
* Copyright (c) 2015 OpenALPR Technology, Inc.
|
|
* Open source 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 OPENALPR_ALPRIMPL_H
|
|
#define OPENALPR_ALPRIMPL_H
|
|
|
|
#include <list>
|
|
#include <sstream>
|
|
#include <vector>
|
|
#include <queue>
|
|
|
|
#include "alpr.h"
|
|
#include "config.h"
|
|
|
|
#include "detection/detector.h"
|
|
#include "detection/detectorfactory.h"
|
|
|
|
#include "prewarp.h"
|
|
|
|
#include "licenseplatecandidate.h"
|
|
#include "../statedetection/state_detector.h"
|
|
#include "segmentation/charactersegmenter.h"
|
|
#include "ocr.h"
|
|
|
|
#include "constants.h"
|
|
|
|
#include "cjson.h"
|
|
|
|
#include "pipeline_data.h"
|
|
|
|
#include "prewarp.h"
|
|
|
|
#include <opencv2/core/core.hpp>
|
|
|
|
#include "support/platform.h"
|
|
#include "support/utf8.h"
|
|
|
|
#define DEFAULT_TOPN 25
|
|
#define DEFAULT_DETECT_REGION false
|
|
|
|
#define ALPR_NULL_PTR 0
|
|
|
|
namespace alpr
|
|
{
|
|
|
|
struct AlprFullDetails
|
|
{
|
|
std::vector<PlateRegion> plateRegions;
|
|
AlprResults results;
|
|
};
|
|
|
|
struct AlprRecognizers
|
|
{
|
|
Detector* plateDetector;
|
|
StateDetector* stateDetector;
|
|
OCR* ocr;
|
|
};
|
|
|
|
class AlprImpl
|
|
{
|
|
|
|
public:
|
|
AlprImpl(const std::string country, const std::string configFile = "", const std::string runtimeDir = "");
|
|
virtual ~AlprImpl();
|
|
|
|
AlprFullDetails recognizeFullDetails(cv::Mat img, std::vector<cv::Rect> regionsOfInterest);
|
|
|
|
AlprResults recognize( std::vector<char> imageBytes );
|
|
AlprResults recognize( std::vector<char> imageBytes, std::vector<AlprRegionOfInterest> regionsOfInterest );
|
|
AlprResults recognize( unsigned char* pixelData, int bytesPerPixel, int imgWidth, int imgHeight, std::vector<AlprRegionOfInterest> regionsOfInterest );
|
|
AlprResults recognize( cv::Mat img );
|
|
AlprResults recognize( cv::Mat img, std::vector<cv::Rect> regionsOfInterest );
|
|
|
|
AlprFullDetails analyzeSingleCountry(cv::Mat colorImg, cv::Mat grayImg, std::vector<cv::Rect> regionsOfInterest);
|
|
|
|
void setCountry(std::string country);
|
|
void setPrewarp(std::string prewarp_config);
|
|
void setMask(unsigned char* pixelData, int bytesPerPixel, int imgWidth, int imgHeight);
|
|
|
|
void setDetectRegion(bool detectRegion);
|
|
void setTopN(int topn);
|
|
void setDefaultRegion(std::string region);
|
|
|
|
static std::string toJson( const AlprResults results );
|
|
static AlprResults fromJson(std::string json);
|
|
static std::string getVersion();
|
|
|
|
static cJSON* createJsonObj(const AlprPlateResult* result);
|
|
|
|
Config* config;
|
|
|
|
bool isLoaded();
|
|
|
|
private:
|
|
|
|
std::map<std::string, AlprRecognizers> recognizers;
|
|
|
|
PreWarp* prewarp;
|
|
|
|
int topN;
|
|
bool detectRegion;
|
|
std::string defaultRegion;
|
|
|
|
void loadRecognizers();
|
|
|
|
cv::Mat getCharacterTransformMatrix(PipelineData* pipeline_data );
|
|
std::vector<AlprCoordinate> getCharacterPoints(cv::Rect char_rect, cv::Mat transmtx);
|
|
std::vector<cv::Rect> convertRects(std::vector<AlprRegionOfInterest> regionsOfInterest);
|
|
|
|
};
|
|
}
|
|
|
|
|
|
#endif // OPENALPR_ALPRIMPL_H
|