You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
20 lines
472 B
C++
20 lines
472 B
C++
#include "MI_Interface.h"
|
|
#include "MF_Resnet34Infer.h"
|
|
#include "MF_Yolov8Infer.h"
|
|
|
|
|
|
MI_VisionInterface* getInterfacePtr(const trtUtils::InitParameter& _params)
|
|
{
|
|
switch (_params.m_modelType)
|
|
{
|
|
case trtUtils::ME_ModelType::E_RESNET34:
|
|
return new MF_Resnet34Infer(_params);
|
|
case trtUtils::ME_ModelType::E_RESNET50:
|
|
return new MF_Resnet34Infer(_params);
|
|
case trtUtils::ME_ModelType::E_YOLOV8:
|
|
return new MF_Yolov8Infer(_params);
|
|
default:
|
|
return nullptr;
|
|
}
|
|
}
|