缺省input_shape参数导致转换失败
可能原因
分析上述日志信息,可能存在以下故障原因:
原始的TensorFlow模型输入是动态shape,如“?×299×299×3”,转换模型的时候缺省了input_shape选项导致报错。
解决措施
- 使用Netron工具查看模型,确认模型输入的shape。
- 对于动态shape的模型,转换时必须加上input_shape参数。命令行举例:
atc --model=./resnetv2.pb --framework=3 --input_shape="input_tensor:8,299,299,3" --input_format="NHWC" --output=./resnetv2 --soc_version=<soc_version>
父主题: FAQ