diff --git a/AIEngineModule/include_base/AI_Factory.h b/AIEngineModule/include_base/AI_Factory.h index ead0eb6..de005b8 100644 --- a/AIEngineModule/include_base/AI_Factory.h +++ b/AIEngineModule/include_base/AI_Factory.h @@ -181,6 +181,8 @@ public: std::shared_ptr AI_defect_RE_POL; // L127 L255 std::shared_ptr AI_defect_RE_AD; // L127 L255 + std::shared_ptr AI_defect_RE_RGBHGRAY; // L127 L255 + std::shared_ptr AI_defect_Edge_Big; // L127 L255 std::shared_ptr AI_defect_Edge_Samll; // L127 L255 diff --git a/AIEngineModule/src/AI_Factory.cpp b/AIEngineModule/src/AI_Factory.cpp index 3d41a62..dfdbbe5 100644 --- a/AIEngineModule/src/AI_Factory.cpp +++ b/AIEngineModule/src/AI_Factory.cpp @@ -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)