|
|
|
|
@ -187,6 +187,17 @@ int AIFactory::InitALLAIModle(GPU_Config gupconfig)
|
|
|
|
|
boe_config.inputType = AIModel_Base::Input_CHW;
|
|
|
|
|
AI_defect_RE_AD->Init(boe_config);
|
|
|
|
|
}
|
|
|
|
|
// RGBHGRAY检测
|
|
|
|
|
if (!AI_defect_RE_RGBHGRAY)
|
|
|
|
|
{
|
|
|
|
|
AI_defect_RE_RGBHGRAY = AIModel_Base::GetInstance();
|
|
|
|
|
AIModel_Base::AIModelRun_Config boe_config;
|
|
|
|
|
boe_config.gpuconfig.copy(gupconfig);
|
|
|
|
|
boe_config.strPath = "/home/aidlux/BOE/UseModel_FOG/FOG_RGBHGRAY.engine";
|
|
|
|
|
boe_config.strName = "RE_RGBHGRAY";
|
|
|
|
|
boe_config.inputType = AIModel_Base::Input_CHW;
|
|
|
|
|
AI_defect_RE_RGBHGRAY->Init(boe_config);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 边缘模型
|
|
|
|
|
if (!AI_defect_Edge_Big)
|
|
|
|
|
|