From 15341944db722076f3ac3299940e7414a0a09d17 Mon Sep 17 00:00:00 2001 From: gitleej <31566147+gitleej@users.noreply.github.com> Date: Wed, 26 Jul 2023 20:16:33 +0800 Subject: [PATCH] Update model_export_cn.md (#3378) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 添加导出dynamic_shape.pdtxt操作步骤 --- docs/model_export_cn.md | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/docs/model_export_cn.md b/docs/model_export_cn.md index 9c49c0f0c4..b45db4464b 100644 --- a/docs/model_export_cn.md +++ b/docs/model_export_cn.md @@ -42,13 +42,22 @@ python tools/export.py \ 注意: * 如果部署模型时,出现和shape相关的问题,请尝试指定input_shape。 -## 3. 预测模型文件 +## 3. 导出dynamic_shape.pdtxt(可选,当使用TensorRT加速,并开启dynamic_shape时,可直接加载该文件) +```shell +python deploy/python/collect_dynamic_shape.py \ + --config output/inference_model/deploy.yaml \ + --image_path path/to/test/image \ + --dynamic_shape_path output/inference_model/dynamic_shape.pbtxt +``` + +## 4. 预测模型文件 如下是导出的预测模型文件。 ```shell output/inference_model ├── deploy.yaml # 部署相关的配置文件,主要说明数据预处理方式等信息 + ├── dynamic_shape.pdtxt # 用于TensorRT加速的dynamic shape信息 ├── model.pdmodel # 预测模型的拓扑结构文件 ├── model.pdiparams # 预测模型的权重文件 └── model.pdiparams.info # 参数额外信息,一般无需关注