文档首页/ 华为HiLens/ 常见问题/ 技能开发/ 模型转换时,如何确定输入张量形状?
更新时间:2022-02-22 GMT+08:00

模型转换时,如何确定输入张量形状?

“输入张量形状”指要添加张量形状,即模型输入数据的shape格式为NHWC,如“input_name:1,224,224,3”,必填项。“input_name”必须是转换前的网络模型中的节点名称。当模型存在动态shape输入时必须提供。例如“input_name1:?,h,w,c”,该参数必填,其中“?”为batch数,表示1次处理的图片数量,需要根据实际情况填写,用于将动态shape的原始模型转换为固定shape的离线模型。

如果存在多个输入,请以分号(;)隔开。

“pb”模型转“om”模型时的张量形状就是“pb”模型的输入节点和shape,例如“images:1,224,224,3”