20 lines
756 B
C++
20 lines
756 B
C++
#include <handleAi/handleAiModel.h>
|
|
#include <appConfig/appConfig.h>
|
|
#include <appCore.hpp>
|
|
|
|
handleAiModelManager::handleAiModelManager()
|
|
{
|
|
_core._appConfig.productDatasMutex.lock();
|
|
this->basePath = _core._appConfig.handleAiModelPath + "/";
|
|
for (auto& productData : _core._appConfig.productDatas)
|
|
{
|
|
insertDetModelPath(productData.handleAiData.detModelName, productData.handleAiData.detGpuIndex);
|
|
for (auto& handleAiData : productData.handleAiData.detResultHandle)
|
|
{
|
|
|
|
insertSegModelPath(handleAiData.segModelName, handleAiData.segGpuIndex);
|
|
insertClsModelPath(handleAiData.clsModelName, handleAiData.clsGpuIndex);
|
|
}
|
|
}
|
|
_core._appConfig.productDatasMutex.unlock();
|
|
} |