专栏感知Pointpillars导出不含预处理模型的配置说明

Pointpillars导出不含预处理模型的配置说明

芯链情报局2023-11-16
83
0

前言

对于PP参考算法,导出的模型默认有预处理,如果您需要导出不含预处理的PP模型,可以按照以下步骤修改配置。这里以J5 OE 1.1.62为例介绍修改流程,不同版本可能会有微小区别。

Config

horizon_model_train_sample/scripts/configs/detection/pointpillars/pointpillars_kitti_car.py

注释deploy_model中的pre_process部分:

将deploy_inputs修改为PFN输入:

structure

/usr/local/lib/python3.8/dist-packages/hat/models/structures/detectors/pointpillar.py

保留features和coors:

PFNLayer

/usr/local/lib/python3.8/dist-packages/hat/models/task_modules/lidar/pillar_encoder.py

做出以下修改:

compile

运行以下命令编译不带预处理的PP模型:


感知
征程5
评论0
0/1000