You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
1.2 KiB
C++
40 lines
1.2 KiB
C++
//
|
|
// Created by 庾金科 on 16/10/2017.
|
|
//
|
|
|
|
#ifndef SWIFTPR_PLATESEGMENTATION_H
|
|
#define SWIFTPR_PLATESEGMENTATION_H
|
|
|
|
#include "opencv2/opencv.hpp"
|
|
#include <opencv2/dnn.hpp>
|
|
#include "PlateInfo.h"
|
|
|
|
namespace pr{
|
|
|
|
|
|
class PlateSegmentation{
|
|
public:
|
|
const int PLATE_NORMAL = 6;
|
|
const int PLATE_NORMAL_GREEN = 7;
|
|
const int DEFAULT_WIDTH = 20;
|
|
PlateSegmentation(std::string phototxt,std::string caffemodel);
|
|
PlateSegmentation(){}
|
|
void segmentPlatePipline(PlateInfo &plateInfo,int stride,std::vector<cv::Rect> &Char_rects);
|
|
|
|
void segmentPlateBySlidingWindows(cv::Mat &plateImage,int windowsWidth,int stride,cv::Mat &respones);
|
|
void templateMatchFinding(const cv::Mat &respones,int windowsWidth,std::pair<float,std::vector<int> > &candidatePts);
|
|
void refineRegion(cv::Mat &plateImage,const std::vector<int> &candidatePts,const int padding,std::vector<cv::Rect> &rects);
|
|
void ExtractRegions(PlateInfo &plateInfo,std::vector<cv::Rect> &rects);
|
|
cv::Mat classifyResponse(const cv::Mat &cropped);
|
|
private:
|
|
cv::dnn::Net net;
|
|
|
|
|
|
// RefineRegion()
|
|
|
|
};
|
|
|
|
}//namespace pr
|
|
|
|
#endif //SWIFTPR_PLATESEGMENTATION_H
|