Apollo之怎么在预测模块中添加新评估器

2023-04-07 07:29:00 模块 添加 评估

在Apollo预测模块中添加新评估器有两种方法:

一种是通过Apollo的配置文件来添加,另一种是通过代码来添加。

通过配置文件来添加评估器需要在配置文件中添加以下内容:

evaluator_config { module: "new_evaluator" //模块名称,可以是现有的模块也可以是自定义的模块。 evaluator_name: "new_evaluator" //评估器名称 output_blob_names: ["new_evaluator_output"] //输出blob的名称 input_blob_names: ["new_evaluator_input"] //输入blob的名称 }

通过代码来添加评估器需要在预测模块的代码中添加以下内容:

class NewEvaluator : public Evaluator { public: explicit NewEvaluator(const std::string& name) : Evaluator(name) {} virtual ~NewEvaluator() {} virtual bool Init(const std::map& params) { //初始化评估器 //TODO return true; } virtual bool Evaluate(const std::vector& inputs, const std::vector& outputs) { //执行评估 //TODO return true; } }; REGISTER_EVALUATOR(NewEvaluator); //注册评估器

在预测模块的代码中添加以上内容之后,需要重新编译预测模块。

相关文章