From f5748700b79fa159068ff9e30f57cc299d34c5e4 Mon Sep 17 00:00:00 2001 From: liusiyang Date: Mon, 1 Jun 2026 09:24:42 +0800 Subject: [PATCH] =?UTF-8?q?add=20=E6=B7=BB=E5=8A=A0RGBHGRAY=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- AIEngineModule/include_base/AI_Factory.h | 2 ++ AIEngineModule/src/AI_Factory.cpp | 11 +++++++++++ 2 files changed, 13 insertions(+) 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)