更新时间: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”。
父主题: 技能开发