28 lines
771 B
C++
28 lines
771 B
C++
#include <string>
|
|
#include <vector>
|
|
#include <opencv.hpp>
|
|
#include <unordered_map>
|
|
struct detResultHandleStruct
|
|
{
|
|
std::vector<int> detResultIndex;//本产品对应det模型训练时的序号
|
|
|
|
std::string segModelName;//重复时将不会重复初始化
|
|
std::vector<int>segNgIndex;//NG的将会进入cls检测
|
|
int segGpuIndex = 0;
|
|
|
|
std::string clsModelName;//重复时将不会重复初始化
|
|
std::vector<int>clsNgIndex;//NG判断
|
|
int clsGpuIndex = 0;
|
|
|
|
//int为Cls的分类结果 string为文本 Scalar为颜色
|
|
std::unordered_map<int, std::pair<std::string, cv::Scalar>>clsResultHandle;//结果绘制
|
|
};
|
|
struct handleAiSimpleDet
|
|
{
|
|
//最后调用模型时是使用路径去查找，请保证规则一致
|
|
std::string detModelName;//重复时将不会重复初始化
|
|
int detGpuIndex = 0;
|
|
|
|
std::vector<detResultHandleStruct>detResultHandle;
|
|
};
|
|
#pragma once |