apollo::prediction::SemanticLstmPedestrianCpuTorch类 参考

#include <semantic_lstm_pedestrian_torch_model.h>

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

apollo::prediction::SemanticLstmPedestrianCpuTorch 的协作图:

Public 成员函数
	SemanticLstmPedestrianCpuTorch ()
	~SemanticLstmPedestrianCpuTorch ()
virtual bool	Init ()
	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)
	performing network inference
virtual bool	LoadModel ()
	load the model from file
virtual void	Destory ()
	free all memory requested, gpu or cpu
Public 成员函数 继承自 apollo::prediction::ModelBase
	ModelBase ()
	~ModelBase ()

额外继承的成员函数
Public 属性 继承自 apollo::prediction::ModelBase
std::string	model_path_
uint8_t	init_ = 0

详细描述

在文件 semantic_lstm_pedestrian_torch_model.h 第 30 行定义.

构造及析构函数说明

SemanticLstmPedestrianCpuTorch()

apollo::prediction::SemanticLstmPedestrianCpuTorch::SemanticLstmPedestrianCpuTorch ( )

inline

在文件 semantic_lstm_pedestrian_torch_model.h 第 32 行定义.

{}

~SemanticLstmPedestrianCpuTorch()

apollo::prediction::SemanticLstmPedestrianCpuTorch::~SemanticLstmPedestrianCpuTorch ( )

inline

在文件 semantic_lstm_pedestrian_torch_model.h 第 33 行定义.

{Destory();}

成员函数说明

Destory()

void apollo::prediction::SemanticLstmPedestrianCpuTorch::Destory ( )

virtual

free all memory requested, gpu or cpu

返回

memory release result, true for success

实现了 apollo::prediction::ModelBase.

在文件 semantic_lstm_pedestrian_torch_model.cc 第 116 行定义.

{}

Inference()

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

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::ModelBase.

在文件 semantic_lstm_pedestrian_torch_model.cc 第 82 行定义.

                                                               {
  ACHECK(input_size == input_buffer.size() && input_size == 3);
  ACHECK(output_size == output_buffer->size() && output_size == 1);
 
  if (init_ == 0) {
    Init();
  }
 
  auto device = torch::Device(torch::kCPU);
  if (torch::cuda::is_available()) {
    device = torch::Device(torch::kCUDA);
  }
  torch::Tensor img_tensor =
      torch::from_blob(input_buffer[0], {1, 3, 224, 224});
  torch::Tensor obstacle_pos = torch::from_blob(input_buffer[1], {1, 20, 2});
  torch::Tensor obstacle_pos_step =
      torch::from_blob(input_buffer[2], {1, 20, 2});
 
  std::vector<torch::jit::IValue> torch_inputs;
 
  torch_inputs.push_back(c10::ivalue::Tuple::create(
      {std::move(img_tensor.to(device)), std::move(obstacle_pos.to(device)),
       std::move(obstacle_pos_step.to(device))}));
 
  torch::Tensor torch_output_tensor =
      model_instance_.forward(torch_inputs).toTensor().to(torch::kCPU);
  memcpy((*output_buffer)[0], torch_output_tensor.data_ptr<float>(),
         1 * 30 * 2 * sizeof(float));
 
  return true;
}

Init()

bool apollo::prediction::SemanticLstmPedestrianCpuTorch::Init ( )

virtual

parse model description class and load the model

参数

config_path

model config path

返回

init result, true for success

实现了 apollo::prediction::ModelBase.

在文件 semantic_lstm_pedestrian_torch_model.cc 第 31 行定义.

                                          {
  ModelConf model_config;
  int status;
 
  if (init_ != 0) {
    return true;
  }
 
  std::string class_name =
      abi::__cxa_demangle(typeid(*this).name(), 0, 0, &status);
 
  std::string default_config_path =
      apollo::cyber::plugin_manager::PluginManager::Instance()
          ->GetPluginConfPath<ModelBase>(class_name,
                                         "conf/default_conf.pb.txt");
 
  if (!cyber::common::GetProtoFromFile(default_config_path, &model_config)) {
    AERROR << "Unable to load model conf file: " << default_config_path;
    return false;
  }
  model_path_ = model_config.model_path();
  init_ = 1;
 
  return LoadModel();
}

LoadModel()

bool apollo::prediction::SemanticLstmPedestrianCpuTorch::LoadModel ( )

virtual

load the model from file

返回

loading result, true for success

实现了 apollo::prediction::ModelBase.

在文件 semantic_lstm_pedestrian_torch_model.cc 第 57 行定义.

                                               {
  auto device = torch::Device(torch::kCPU);
  if (torch::cuda::is_available()) {
    device = torch::Device(torch::kCUDA);
  }
 
  model_instance_ = torch::jit::load(model_path_, device);
  torch::set_num_threads(1);
 
  // Fake intput for the first frame
  torch::Tensor img_tensor = torch::randn({1, 3, 224, 224});
  torch::Tensor obstacle_pos = torch::randn({1, 20, 2});
  torch::Tensor obstacle_pos_step = torch::randn({1, 20, 2});
  std::vector<torch::jit::IValue> torch_inputs;
  torch::Tensor torch_default_output_tensor;
 
  torch_inputs.push_back(c10::ivalue::Tuple::create(
      {std::move(img_tensor.to(device)), std::move(obstacle_pos.to(device)),
       std::move(obstacle_pos_step.to(device))}));
 
  // warm up to avoid very slow first inference later
  WarmUp(torch_inputs, &model_instance_, &torch_default_output_tensor);
  return true;
}

---

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

• modules/prediction/evaluator/model_manager/model/semantic_lstm_pedestrian_torch_cpu/semantic_lstm_pedestrian_torch_model.h
• modules/prediction/evaluator/model_manager/model/semantic_lstm_pedestrian_torch_cpu/semantic_lstm_pedestrian_torch_model.cc