apollo_logo
2
0

CenterPoint训练部署说明

本教程将介绍如何使用apolloscape数据训练CenterPoint,同时部署到Apollo中运行查看结果。

训练

我们提供了两种训练方式:

介绍

文档链接

本地训练

用户在本地完成环境配置、数据下载转换、模型训练导出的流程

AI Studio训练

用户登录AI Studio官网,并在我们提供的项目中直接开始训练

(环境、数据已处理好,开发者可快速展开训练)


训练参数说明,训练指导建议参考文档CenterPoint训练指导

部署

按照上述步骤得到可部署文件,如下所示:

|-- CenterPoint
|-- apollo_deploy.yaml
|-- centerpoint.pdiparams
|-- centerpoint.pdiparams.info
|-- centerpoint.pdmodel
  • 将上述4个文件放置到/apollo/modules/perception/data/models/center_point_paddle路径下,用于本地调试;
  • 将上述4个文件放置到/apollo_workspace/modules/perception/data/models/center_point_paddle路径下,用于提交代码;

修改模型输出名称

  • 在本地Ubuntu安装visualdl;
python -m pip install visualdl -i https://mirror.baidu.com/pypi/simple

上传centerpoint.pdmodel,点击输出(下图中的黄色框)查看输出名称

该例子中输出名称分别为_generated_var_9,_generated_var_10,_generated_var_11

在center_point_param.pb.txt中进行修改:

outputs {name: "_generated_var_9"}
outputs {name: "_generated_var_10"}
outputs {name: "_generated_var_11"}

到此为止,模型的部署流程完成。

原创声明,本文由作者授权发布于Apollo开发者社区,未经许可,不得转载。
发表评论已发表 0 条评论
登录后可评论,请前往 登录
暂无评论~快去发表自己的独特见解吧!
目录
训练
部署