#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;
	}
}