Vitis AI TensorFlow量化时遇input_1节点未找到错误的技术问询
量化时输入节点不存在的原因分析
冻结图路径变量未正确解析
你的代码中FROZEN_GRAPH='${MODEL_PATH}/frozen_graph.pb'使用了单引号,shell中单引号会阻止变量替换,导致实际加载的路径是字符串${MODEL_PATH}/frozen_graph.pb,而非预期的./trained_models/saved_model/frozen_graph.pb。加载错误的文件(甚至找不到文件时可能加载空图),自然会提示输入节点不存在。冻结图导出不完整
你用Netron查看的是SavedModel,但量化用的是冻结图frozen_graph.pb。如果导出冻结图的过程有误(比如导出时指定的输入/输出节点不正确,或者没有完整保存所有节点),会导致冻结图中缺失input_1节点。节点名称存在差异
Netron显示的节点名可能是显示名称,而非图中实际使用的节点全名。比如部分模型导出时会自动添加命名空间前缀(如model/input_1),此时你指定的input_1就无法匹配到实际节点。可以通过以下命令查看冻结图中的所有节点名,确认实际的输入节点:vai_q_tensorflow inspect --input_frozen_graph /path/to/your/frozen_graph.pb输入节点参数格式问题
虽然你只有一个输入节点,但需确认--input_nodes参数的格式是否符合要求。若节点名包含特殊字符或空格,需要用引号包裹,但你这里的情况更可能是前面的路径问题。
内容的提问来源于stack exchange,提问作者Kevini Perera
相关产品推荐
相关产品推荐

