Ubuntu
74 lines
2.5 KiB
C++
74 lines
2.5 KiB
C++
#pragma once
|
|
|
|
#include "Types.h"
|
|
#include "ColorReader.h"
|
|
#include "PaperConfig.h"
|
|
#include "GlobalParams.h"
|
|
|
|
class PaperRecognizer {
|
|
public:
|
|
PaperRecognizer();
|
|
~PaperRecognizer();
|
|
|
|
// 识别试纸
|
|
RecognitionResult recognize(const cv::Mat& image);
|
|
|
|
// 获取裁剪后的试纸图
|
|
cv::Mat getPaperImg() const;
|
|
|
|
// 读取所有色块颜色
|
|
std::vector<ColorResult> readAllColorBlocks(const cv::Mat& bgrImage);
|
|
|
|
private:
|
|
// 定位块配对
|
|
struct LocatorPair {
|
|
bool isValid;
|
|
LocatorInfo* big;
|
|
LocatorInfo* small;
|
|
LocatorPair() : isValid(false), big(nullptr), small(nullptr) {}
|
|
};
|
|
|
|
void preprocessImage(const cv::Mat& image);
|
|
bool detectLocators();
|
|
void findCandidateContours(std::vector<std::vector<cv::Point>>& contours);
|
|
bool filterLocatorCandidates(const std::vector<std::vector<cv::Point>>& contours);
|
|
std::vector<std::vector<LocatorInfo*>> groupCandidatesByY(std::vector<LocatorInfo>& candidates);
|
|
LocatorPair findValidLocatorPairInGroup(std::vector<LocatorInfo*>& group);
|
|
bool validateLocatorPair();
|
|
double calculateAspectRatio(const cv::RotatedRect& rect);
|
|
bool isPotentialLocator(const cv::RotatedRect& rect, double area, double& aspectRatio);
|
|
void sortLocatorsByPosition();
|
|
BigLocatorType determineBigLocatorType(double areaRatio);
|
|
bool matchPaperSpec();
|
|
void calculateColorBlockCenters();
|
|
BlurLevel assessLocatorBlur(const cv::Mat& gray, const cv::RotatedRect& locator, double& score);
|
|
void drawAnnotations();
|
|
void savePaperImg();
|
|
LocatorColor determineLocatorColor(const cv::Rect& roi);
|
|
LocatorColor determineLocatorColorByCenter(const cv::RotatedRect& rotatedRect);
|
|
cv::Scalar getCenterRegionColor(const cv::RotatedRect& rotatedRect);
|
|
cv::Scalar getAverageColor(const cv::Rect& roi);
|
|
double measureSaturation(const cv::Mat& hsvRoi);
|
|
double calculateColorDistance(const cv::Scalar& c1, const cv::Scalar& c2);
|
|
|
|
cv::Mat originalImage_; // 原图
|
|
cv::Mat processImage_; // 处理用图像
|
|
cv::Mat hsvImage_; // HSV图像
|
|
cv::Mat grayImage_; // 灰度图像
|
|
cv::Mat paperImg_; // 裁剪后的试纸图片（RGBA格式）
|
|
|
|
bool hasLocator1_;
|
|
bool hasLocator2_;
|
|
LocatorInfo locator1_;
|
|
LocatorInfo locator2_;
|
|
LocatorInfo* bigLocator_;
|
|
LocatorInfo* smallLocator_;
|
|
|
|
std::vector<cv::Point2f> centers_;
|
|
RecognitionResult result_;
|
|
const PaperConfig* config_;
|
|
ColorReader colorReader_;
|
|
|
|
size_t candidateCount_;
|
|
};
|