Update README.md

Fix doc
own
Lin Manhui 2 years ago committed by GitHub
parent 1598fa951e
commit 9353d18ae8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      deploy/export/README.md

@ -47,7 +47,7 @@ python deploy/export/export_model.py --model_dir=./output/deeplabv3p/best_model/
| --save_dir | 导出的部署格式模型存储路径,例如`./inference_model/`。 | | --save_dir | 导出的部署格式模型存储路径,例如`./inference_model/`。 |
| --fixed_input_shape | 固定导出模型的输入张量形状。默认值为None,表示使用任务默认输入张量形状。 | | --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]`。
完整命令示例: 完整命令示例:

Loading…
Cancel
Save