apollo::prediction::ModelBase类 参考

#include <model_base.h>

类 apollo::prediction::ModelBase 继承关系图:

apollo::prediction::ModelBase 的协作图:

Public 成员函数
	ModelBase ()
	~ModelBase ()
virtual bool	Init ()=0
	parse model description class and load the model
virtual bool	Inference (const std::vector< void * > &input_buffer, unsigned int input_size, std::vector< void * > *output_buffer, unsigned int output_size)=0
	performing network inference
virtual bool	LoadModel ()=0
	load the model from file
virtual void	Destory ()=0
	free all memory requested, gpu or cpu

Public 属性
std::string	model_path_
uint8_t	init_ = 0

详细描述

在文件 model_base.h 第 25 行定义.

构造及析构函数说明

ModelBase()

apollo::prediction::ModelBase::ModelBase ( )

inline

在文件 model_base.h 第 27 行定义.

{}

~ModelBase()

apollo::prediction::ModelBase::~ModelBase ( )

inline

在文件 model_base.h 第 28 行定义.

{}

成员函数说明

Destory()

virtual void apollo::prediction::ModelBase::Destory ( )

pure virtual

free all memory requested, gpu or cpu

返回

memory release result, true for success

在 apollo::prediction::MultiAgentPedestrianCpuTorch, apollo::prediction::MultiAgentPedestrianGpuTorch, apollo::prediction::MultiAgentVehicleCpuTorch, apollo::prediction::MultiAgentVehicleGpuTorch, apollo::prediction::SemanticLstmPedestrianCpuTorch, apollo::prediction::SemanticLstmPedestrianGpuTorch, apollo::prediction::SemanticLstmVehicleCpuTorch, apollo::prediction::SemanticLstmVehicleGpuTorch, apollo::prediction::MultiAgentPedestrianTensorrt, apollo::prediction::MultiAgentVehicleTensorrt, apollo::prediction::SemanticLstmPedestrianTensorrt , 以及 apollo::prediction::SemanticLstmVehicleTensorrt 内被实现.

Inference()

virtual bool apollo::prediction::ModelBase::Inference ( const std::vector< void * > &  input_buffer, unsigned int  input_size, std::vector< void * > *  output_buffer, unsigned int  output_size  )

pure virtual

performing network inference

参数

input_buffer	vector of input tensor
input_size	size of input_buffer
output_buffer	vector of output tensor
output_size	size of output_buffer

返回

init result, true for success

在 apollo::prediction::MultiAgentPedestrianCpuTorch, apollo::prediction::MultiAgentPedestrianGpuTorch, apollo::prediction::MultiAgentVehicleCpuTorch, apollo::prediction::MultiAgentVehicleGpuTorch, apollo::prediction::SemanticLstmPedestrianCpuTorch, apollo::prediction::SemanticLstmPedestrianGpuTorch, apollo::prediction::SemanticLstmVehicleCpuTorch, apollo::prediction::SemanticLstmVehicleGpuTorch, apollo::prediction::MultiAgentPedestrianTensorrt, apollo::prediction::MultiAgentVehicleTensorrt, apollo::prediction::SemanticLstmPedestrianTensorrt , 以及 apollo::prediction::SemanticLstmVehicleTensorrt 内被实现.

Init()

virtual bool apollo::prediction::ModelBase::Init ( )

pure virtual

parse model description class and load the model

参数

config_path

the model config

返回

init result, true for success

在 apollo::prediction::MultiAgentPedestrianCpuTorch, apollo::prediction::MultiAgentPedestrianGpuTorch, apollo::prediction::MultiAgentVehicleCpuTorch, apollo::prediction::MultiAgentVehicleGpuTorch, apollo::prediction::SemanticLstmPedestrianCpuTorch, apollo::prediction::SemanticLstmPedestrianGpuTorch, apollo::prediction::SemanticLstmVehicleCpuTorch, apollo::prediction::SemanticLstmVehicleGpuTorch, apollo::prediction::MultiAgentPedestrianTensorrt, apollo::prediction::MultiAgentVehicleTensorrt, apollo::prediction::SemanticLstmPedestrianTensorrt , 以及 apollo::prediction::SemanticLstmVehicleTensorrt 内被实现.

LoadModel()

virtual bool apollo::prediction::ModelBase::LoadModel ( )

pure virtual

load the model from file

返回

loading result, true for success

在 apollo::prediction::MultiAgentPedestrianCpuTorch, apollo::prediction::MultiAgentPedestrianGpuTorch, apollo::prediction::MultiAgentVehicleCpuTorch, apollo::prediction::MultiAgentVehicleGpuTorch, apollo::prediction::SemanticLstmPedestrianCpuTorch, apollo::prediction::SemanticLstmPedestrianGpuTorch, apollo::prediction::SemanticLstmVehicleCpuTorch, apollo::prediction::SemanticLstmVehicleGpuTorch, apollo::prediction::MultiAgentPedestrianTensorrt, apollo::prediction::MultiAgentVehicleTensorrt, apollo::prediction::SemanticLstmPedestrianTensorrt , 以及 apollo::prediction::SemanticLstmVehicleTensorrt 内被实现.

类成员变量说明

init_

uint8_t apollo::prediction::ModelBase::init_ = 0

在文件 model_base.h 第 67 行定义.

model_path_

std::string apollo::prediction::ModelBase::model_path_

在文件 model_base.h 第 66 行定义.

---

该类的文档由以下文件生成:

• modules/prediction/evaluator/model_manager/model/model_base.h