You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
98 lines
3.2 KiB
98 lines
3.2 KiB
/*
|
|
* @Author: xiewenji 527774126@qq.com
|
|
* @Date: 2025-09-03 10:33:48
|
|
* @LastEditors: xiewenji 527774126@qq.com
|
|
* @LastEditTime: 2025-09-14 20:28:10
|
|
* @FilePath: /AI_SO_Test/AIEngineModule/src/AI_Factory.cpp
|
|
* @Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
|
|
*/
|
|
#include "AI_Factory.h"
|
|
|
|
#include "AIModel_Impl.h"
|
|
|
|
std::shared_ptr<AIModel_Base> AIModel_Base::GetInstance()
|
|
{
|
|
std::shared_ptr<AIModel_Base> instance = std::make_shared<AIModel_Impl>();
|
|
return instance;
|
|
}
|
|
|
|
AIFactory::AIFactory()
|
|
{
|
|
}
|
|
|
|
AIFactory::~AIFactory()
|
|
{
|
|
}
|
|
std::shared_ptr<AIFactory> AIFactory::GetInstance()
|
|
{
|
|
static std::shared_ptr<AIFactory> instance = std::make_shared<AIFactory>();
|
|
return instance;
|
|
}
|
|
|
|
int AIFactory::InitALLAIModle(GPU_Config gupconfig)
|
|
{
|
|
if (gupconfig.GetNum() == 0)
|
|
{
|
|
printf("GPU Num Error \n");
|
|
return 1;
|
|
}
|
|
|
|
if (!Defect)
|
|
{
|
|
Defect = AIModel_Base::GetInstance();
|
|
AIModel_Base::AIModelRun_Config edge_config;
|
|
edge_config.gpuconfig.copy(gupconfig);
|
|
edge_config.strPath = "/home/aidlux/BOE/UseModel_ZB/BM/DefectLight.engine";
|
|
edge_config.strName = "Defect";
|
|
edge_config.inputType = AIModel_Base::Input_CHW;
|
|
edge_config.Stream_num = 4;
|
|
Defect->Init(edge_config);
|
|
}
|
|
if (!Class)
|
|
{
|
|
Class = AIModel_Base::GetInstance();
|
|
AIModel_Base::AIModelRun_Config jbl_config;
|
|
jbl_config.gpuconfig.copy(gupconfig);
|
|
jbl_config.strPath = "/home/aidlux/BOE/UseModel_ZB/BM/Class.engine";
|
|
jbl_config.inputType = AIModel_Base::Input_CHW;
|
|
jbl_config.strName = "Class";
|
|
jbl_config.IsClass = true;
|
|
Class->Init(jbl_config);
|
|
}
|
|
if (!Tag_Loc)
|
|
{
|
|
Tag_Loc = AIModel_Base::GetInstance();
|
|
AIModel_Base::AIModelRun_Config edge_config;
|
|
edge_config.gpuconfig.copy(gupconfig);
|
|
edge_config.strPath = "/home/aidlux/BOE/UseModel_ZB/BM/BadLabel.engine";
|
|
edge_config.strName = "Tag_Loc";
|
|
edge_config.inputType = AIModel_Base::Input_CHW;
|
|
edge_config.Stream_num = 4;
|
|
Tag_Loc->Init(edge_config);
|
|
}
|
|
if (!Align_Outer)
|
|
{
|
|
Align_Outer = AIModel_Base::GetInstance();
|
|
AIModel_Base::AIModelRun_Config edge_config;
|
|
edge_config.gpuconfig.copy(gupconfig);
|
|
edge_config.strPath = "/home/aidlux/BOE/UseModel_ZB/BM/ProductLoc.engine";
|
|
edge_config.strName = "Align_Outer";
|
|
edge_config.inputType = AIModel_Base::Input_CHW;
|
|
edge_config.Stream_num = 2;
|
|
Align_Outer->Init(edge_config);
|
|
}
|
|
if (!Align_Inner)
|
|
{
|
|
Align_Inner = AIModel_Base::GetInstance();
|
|
AIModel_Base::AIModelRun_Config edge_config;
|
|
edge_config.gpuconfig.copy(gupconfig);
|
|
edge_config.strPath = "/home/aidlux/BOE/UseModel_ZB/BM/ClampLoc.engine";
|
|
edge_config.strName = "Align_Inner";
|
|
edge_config.inputType = AIModel_Base::Input_CHW;
|
|
edge_config.Stream_num = 2;
|
|
Align_Inner->Init(edge_config);
|
|
}
|
|
m_bInitSucc = true;
|
|
return 0;
|
|
}
|