Added pipeline_data class

This commit is contained in:
Matt Hill
2014-07-01 14:14:38 -04:00
parent bab6372258
commit a52fa6d5f9
2 changed files with 61 additions and 0 deletions

View File

@@ -0,0 +1,13 @@
#include "pipeline_data.h"
using namespace cv;
using namespace std;
PipelineData::PipelineData(Mat colorImage, Rect regionOfInterest, Config* config)
{
this->colorImg = colorImage;
cvtColor(this->colorImg, this->grayImg, CV_BGR2GRAY);
this->regionOfInterest = regionOfInterest;
this->config = config;
}

View File

@@ -0,0 +1,48 @@
#ifndef OPENALPR_PIPELINEDATA_H
#define OPENALPR_PIPELINEDATA_H
#include "segmentation/segment.h"
#include "segmentation/segmentationgroup.h"
class PipelineData
{
public:
PipelineData(cv::Mat colorImage, cv::Rect regionOfInterest, Config* config);
virtual ~PipelineData();
// Inputs
Config* config;
cv::Mat colorImg;
cv::Mat grayImg;
cv::Rect regionOfInterest;
cv::Mat crop_gray;
cv::Mat plate_mask;
// Outputs
std::string region_code;
float region_confidence;
float overall_confidence;
std::vector<cv::Mat> thresholds;
// Plate Lines
std::vector<LineSegment> horizontalLines;
std::vector<LineSegment> verticalLines;
// Segmentation
std::vector<Segment> segments;
std::vector<SegmentationGroup> segmentGroups;
// OCR
};
#endif // OPENALPR_PIPELINEDATA_H