AI开发平台ModelArts-转换模板:Tensorflow frozen graph 转 Ascend

时间：2023-11-01 16:21:52

AI开发平台ModelArts

转换Tensorflow框架训练并以“frozen_graph”格式保存的模型，转换后模型可在Ascend上运行。

表1 Tensorflow frozen graph 转 Ascend的高级选项
参数名称	参数解释
“输入张量形状”	模型输入数据的shape，输入数据格式为NHWC，如“input_name:1,224,224,3”，必填项。“input_name”必须是转换前的网络模型中的节点名称。当模型存在动态shape输入时必须提供。例如“input_name1:?,h,w,c”，该参数必填，其中“?”为batch数，表示1次处理的图片数量，需要根据实际情况填写，用于将动态shape的原始模型转换为固定shape的离线模型。目前不支持批量特性，转换输入张量形状batch只能为1。