mirror of
https://github.com/kerberos-io/openalpr-base.git
synced 2025-10-05 13:16:49 +08:00
66 lines
2.0 KiB
C++
66 lines
2.0 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_PREWARP_H
|
|
#define OPENALPR_PREWARP_H
|
|
|
|
#include "config.h"
|
|
#include "utility.h"
|
|
#include "opencv2/imgproc/imgproc.hpp"
|
|
#include "detection/detector_types.h"
|
|
|
|
namespace alpr
|
|
{
|
|
|
|
class PreWarp {
|
|
public:
|
|
PreWarp(Config* config);
|
|
virtual ~PreWarp();
|
|
|
|
void initialize(std::string prewarp_config);
|
|
void clear();
|
|
|
|
cv::Mat warpImage(cv::Mat image);
|
|
std::vector<cv::Point2f> projectPoints(std::vector<cv::Point2f> points, bool inverse);
|
|
std::vector<cv::Rect> projectRects(std::vector<cv::Rect> rects, int maxWidth, int maxHeight, bool inverse);
|
|
cv::Rect projectRect(cv::Rect rect, int maxWidth, int maxHeight, bool inverse);
|
|
void projectPlateRegions(std::vector<PlateRegion>& plateRegions, int maxWidth, int maxHeight, bool inverse);
|
|
|
|
void setTransform(float w, float h, float rotationx, float rotationy, float rotationz, float panX, float panY, float stretchX, float dist);
|
|
|
|
bool valid;
|
|
|
|
std::string toString();
|
|
|
|
private:
|
|
Config* config;
|
|
cv::Mat transform;
|
|
|
|
cv::Mat getTransform(float w, float h, float rotationx, float rotationy, float rotationz, float panX, float panY, float stretchX, float dist);
|
|
|
|
float w, h, rotationx, rotationy, rotationz, stretchX, dist, panX, panY;
|
|
|
|
};
|
|
|
|
}
|
|
|
|
#endif /* OPENALPR_PREWARP_H */
|
|
|