|
|
/*
|
|
|
* @Author: your name
|
|
|
* @Date: 2022-04-20 15:49:50
|
|
|
* @LastEditTime: 2022-09-23 21:51:58
|
|
|
* @LastEditors: sueRimn
|
|
|
* @Description: 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
|
|
|
* @FilePath: /ZCXD_MonitorPlatform/src/CoreLogicModule/include/CamDeal.h
|
|
|
*/
|
|
|
#ifndef SingleGPU_H_
|
|
|
#define SingleGPU_H_
|
|
|
|
|
|
#include <vector>
|
|
|
#include <thread>
|
|
|
#include <mutex>
|
|
|
#include <string>
|
|
|
#include <iostream>
|
|
|
#include <memory>
|
|
|
#include <opencv2/opencv.hpp>
|
|
|
#include "AICheck.h"
|
|
|
using namespace std;
|
|
|
#define MAX_GPU_NUMBER 4
|
|
|
class AI_SingleGPU
|
|
|
{
|
|
|
public:
|
|
|
AI_SingleGPU(int nGpuIdx);
|
|
|
~AI_SingleGPU();
|
|
|
static AI_SingleGPU *GetInstance(int nGpuIdx);
|
|
|
static AI_SingleGPU *m_pInstance[MAX_GPU_NUMBER];
|
|
|
|
|
|
private:
|
|
|
// 定义一个内部类
|
|
|
class CGarbo
|
|
|
{
|
|
|
public:
|
|
|
CGarbo() {};
|
|
|
~CGarbo()
|
|
|
{
|
|
|
for (int i = 0; i < MAX_GPU_NUMBER; i++)
|
|
|
{
|
|
|
if (nullptr != m_pInstance[i])
|
|
|
{
|
|
|
delete m_pInstance[i];
|
|
|
m_pInstance[i] = nullptr;
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
};
|
|
|
// 定义一个内部类的静态对象
|
|
|
// 当该对象销毁时，顺带就释放myInstance指向的堆区资源
|
|
|
static CGarbo m_garbo;
|
|
|
|
|
|
private:
|
|
|
int m_nGpuIdx;
|
|
|
|
|
|
public:
|
|
|
AI_defect AI_defect_NF; // BOE测试
|
|
|
AI_defect AI_defect_Type2; // 第二个检测模型
|
|
|
AI_defect AI_defect_Cls; // 1023
|
|
|
AI_defect AI_defect_Cls_L0; //
|
|
|
AI_defect AI_defect_YX_1; // L0
|
|
|
AI_defect AI_defect_YX_2; // L127 L255
|
|
|
AI_defect AI_defect_zf; // L127 L255
|
|
|
AI_defect AI_defect_UP; // L127 L255
|
|
|
AI_defect AI_defect_Chess; // L127 L255
|
|
|
AI_defect AI_defect_127Cell; // L127 L255
|
|
|
AI_defect AI_defect_RE_POL; // L127 L255
|
|
|
AI_defect AI_defect_RE_AD; // L127 L255
|
|
|
|
|
|
AI_defect AI_defect_Edge_Big; // L127 L255
|
|
|
AI_defect AI_defect_Edge_Samll; // L127 L255
|
|
|
|
|
|
AI_defect AI_defect_LackPol; // 缺pol检测
|
|
|
AI_defect AI_defect_MarkLine; // MarkLine
|
|
|
};
|
|
|
|
|
|
#endif |
