From 9353d18ae83044adf11736bfaf520d3b9f821e3d Mon Sep 17 00:00:00 2001 From: Lin Manhui Date: Fri, 17 Jun 2022 22:52:21 +0800 Subject: [PATCH] Update README.md Fix doc --- deploy/export/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deploy/export/README.md b/deploy/export/README.md index ae8e7d3..ea03a4b 100644 --- a/deploy/export/README.md +++ b/deploy/export/README.md @@ -47,7 +47,7 @@ python deploy/export/export_model.py --model_dir=./output/deeplabv3p/best_model/ | --save_dir | 导出的部署格式模型存储路径，例如`./inference_model/`。 | | --fixed_input_shape | 固定导出模型的输入张量形状。默认值为None，表示使用任务默认输入张量形状。 | -当使用TensorRT执行模型推理时，需固定模型的输入张量形状。此时，可通过`--fixed_input_shape`选项来指定输入形状，具体有两种形式：`[w,h]`或者`[n,c,w,h]`。例如，指定`--fixed_input_shape`为`[224,224]`时，实际的输入张量形状可视为`[-1,3,224,224]`（-1表示可以为任意正整数，通道数默认为3）；若想同时固定输入数据在batch维度的大小为1、通道数为4，则可将该选项设置为`[1,4,224,224]`。 +当使用TensorRT执行模型推理时，需固定模型的输入张量形状。此时，可通过`--fixed_input_shape`选项来指定输入形状，具体有两种形式：`[w,h]`或者`[n,c,w,h]`。例如，指定`--fixed_input_shape`为`[224,224]`时，实际的输入张量形状可视为`[1,3,224,224]`（batch size默认为1，通道数默认为3）；若需要使模型能够处理任何batch size（需为正整数）的通道数为4的输入，则可将该选项设置为`[-1,4,224,224]`。 完整命令示例：