--input_shape
功能说明
指定模型输入数据的shape。
关联参数
设置shape分档场景,需要配合使用--dynamic_batch_size(设置BatchSize档位)或--dynamic_image_size(设置分辨率档位)或--dynamic_dims(设置指定维度档位)参数。
参数取值
参数值:
- 固定shape:
- 若模型为单个输入,则shape信息为"input_name:n,c,h,w";指定的节点必须放在双引号中。
- 若模型有多个输入,则shape信息--input_shape="input_name1:n1,c1,h1,w1;input_name2:n2,c2,h2,w2";不同输入之间使用英文分号分隔,input_name必须是转换前的网络模型中的节点名称。
- 若原始模型中输入数据的某个或某些维度值不固定,当前支持通过设置shape分档转换模型:包括设置BatchSize档位、设置分辨率档位、设置指定维度档位(最多4维)。
设置--input_shape参数时,将对应维度值设置为-1,同时配合使用--dynamic_batch_size(设置BatchSize档位)或--dynamic_image_size(设置分辨率档位)或--dynamic_dims(设置指定维度档位)参数。
参数值约束:
- 如果模型输入数据为标量(例如BOOL类型),该场景下无需配置对应节点的--input_shape参数。
- 若原始模型中输入数据的某个维度值不固定(例如input_name1:?,h,w,c),通过Netron等可视化软件打开模型之后,输入信息样例如下:
该场景下可以进行如下操作:
- 固定shape,将维度值设置为固定取值,例如,input_name1:1,h,w,c,用于将输入数据某个维度不固定的原始模型转换为固定维度的离线模型。
- 设置shape分档,例如设置为“-1”,与--dynamic_batch_size参数配合使用。
- 若使用该参数时,同时通过--insert_op_conf设置了AIPP功能,则AIPP输出图片的宽和高要在本参数所设置的范围内。
推荐配置及收益
无。
示例
- 固定shape,例如某网络的输入shape信息,输入1:input_0_0 [16,32,208,208],输入2:input_1_0 [16,64,208,208],则--input_shape的配置信息为:
--input_shape="input_0_0:16,32,208,208;input_1_0:16,64,208,208"
- 设置BatchSize档位的示例,请参见--dynamic_batch_size。
- 设置分辨率档位的示例,请参见--dynamic_image_size。
- 设置指定维度档位的示例,请参见--dynamic_dims。
支持的芯片型号
昇腾310B AI处理器
依赖约束
无。
父主题: 输入选项