add 添加RGBHGRAY模型

main
liusiyang 3 weeks ago
parent 92afa850b7
commit f5748700b7

@ -181,6 +181,8 @@ public:
std::shared_ptr<AIModel_Base> AI_defect_RE_POL; // L127 L255 std::shared_ptr<AIModel_Base> AI_defect_RE_POL; // L127 L255
std::shared_ptr<AIModel_Base> AI_defect_RE_AD; // L127 L255 std::shared_ptr<AIModel_Base> AI_defect_RE_AD; // L127 L255
std::shared_ptr<AIModel_Base> AI_defect_RE_RGBHGRAY; // L127 L255
std::shared_ptr<AIModel_Base> AI_defect_Edge_Big; // L127 L255 std::shared_ptr<AIModel_Base> AI_defect_Edge_Big; // L127 L255
std::shared_ptr<AIModel_Base> AI_defect_Edge_Samll; // L127 L255 std::shared_ptr<AIModel_Base> AI_defect_Edge_Samll; // L127 L255

@ -187,6 +187,17 @@ int AIFactory::InitALLAIModle(GPU_Config gupconfig)
boe_config.inputType = AIModel_Base::Input_CHW; boe_config.inputType = AIModel_Base::Input_CHW;
AI_defect_RE_AD->Init(boe_config); 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) if (!AI_defect_Edge_Big)

Loading…
Cancel
Save