Ubuntu
93 lines
3.0 KiB
C++
93 lines
3.0 KiB
C++
#pragma once
|
|
|
|
#include <opencv2/opencv.hpp>
|
|
#include <vector>
|
|
#include <string>
|
|
|
|
// 试纸联数
|
|
enum class BlockItems {
|
|
BI_14 = 14,
|
|
BI_11 = 11,
|
|
BI_8 = 8
|
|
};
|
|
|
|
// 大定位块类型
|
|
enum class BigLocatorType {
|
|
BLT_SHORT = 0, // 10mm
|
|
BLT_LONG = 1 // 15mm
|
|
};
|
|
|
|
// 定位块颜色分类
|
|
enum class LocatorColor {
|
|
UNKNOWN = 0,
|
|
RED = 1,
|
|
GREEN = 2,
|
|
BLUE = 3
|
|
};
|
|
|
|
// 试纸空间类型
|
|
enum class PaperSpatialType {
|
|
PST_UNKNOWN = 0,
|
|
PST_14_S = 1, // 14联, 10mm大定位块
|
|
PST_14_L = 2, // 14联, 15mm大定位块
|
|
PST_11_S = 3, // 11联, 10mm大定位块
|
|
PST_11_L = 4, // 11联, 15mm大定位块
|
|
PST_8_S = 5, // 8联, 10mm大定位块
|
|
PST_8_L = 6 // 8联, 15mm大定位块
|
|
};
|
|
|
|
// 图像清晰度等级
|
|
enum class BlurLevel {
|
|
BLUR_NONE = 0, // 清晰
|
|
BLUR_SLIGHT = 1, // 轻微模糊
|
|
BLUR_SEVERE = 2 // 严重模糊
|
|
};
|
|
|
|
// 定位块信息
|
|
struct LocatorInfo {
|
|
cv::RotatedRect rotatedRect; // 旋转矩形
|
|
cv::Point2f center; // 中心点
|
|
double area; // 面积
|
|
double width; // 宽度（长边）
|
|
double height; // 高度（短边）
|
|
LocatorColor color; // 颜色
|
|
bool isBig; // 是否大定位块
|
|
cv::Scalar avgColor; // 平均RGB颜色
|
|
|
|
LocatorInfo() : area(0), width(0), height(0),
|
|
color(LocatorColor::UNKNOWN), isBig(false) {}
|
|
};
|
|
|
|
// 色块颜色结果
|
|
struct ColorResult {
|
|
int id; // 编号（从靠近小定位块开始）
|
|
double h, s, v; // HSV值
|
|
double l, a, b; // LAB值
|
|
double r, g, b_val; // RGB值
|
|
|
|
ColorResult() : id(0), h(0), s(0), v(0), l(0), a(0), b(0),
|
|
r(0), g(0), b_val(0) {}
|
|
};
|
|
|
|
// 识别结果
|
|
struct RecognitionResult {
|
|
bool success; // 是否成功
|
|
int errorCode; // 错误码
|
|
std::string errorMessage; // 错误信息
|
|
PaperSpatialType paperType; // 试纸类型
|
|
LocatorColor locatorColor; // 定位块颜色
|
|
int itemCount; // 色块数量
|
|
std::vector<cv::Point2f> colorBlockCenters; // 色块中心点
|
|
std::vector<ColorResult> colors; // 颜色结果
|
|
cv::Mat annotatedImage; // 标注后的图片
|
|
bool smallLocatorOnLeft; // 小定位块是否在左边
|
|
BlurLevel blurLevel; // 图像清晰度等级
|
|
double blurScore; // 清晰度分数（方差值）
|
|
|
|
RecognitionResult() : success(false), errorCode(-1),
|
|
paperType(PaperSpatialType::PST_UNKNOWN),
|
|
locatorColor(LocatorColor::UNKNOWN),
|
|
itemCount(0), smallLocatorOnLeft(true),
|
|
blurLevel(BlurLevel::BLUR_NONE), blurScore(0) {}
|
|
};
|