尝试使用Tensorflow 2预测SavedModel时出错

VBK

我正在尝试使用以下代码预测一个保存的模型

 features = np.ones((20, 40, 3), dtype=np.float32)
 features = tf.convert_to_tensor(value, dtype=tf.float32)
 imported_model = tf.saved_model.load(export_dir=os.path.join(os.path.join(model_path, directory)))
 import_fn = imported_model.signatures["serving_default"]
 import_fn(features)

使用Tensorflow 2运行它时出现以下错误。当我使用saved_model_cli时,模型预测工作正常。

tensorflow.python.framework.errors_impl.InvalidArgumentError:  In[0] is not a matrix. Instead it has shape [20,40,3]
     [[node dense/BiasAdd (defined at model_manager.py:54) ]] [Op:__inference_pruned_318590]

保存的cli命令如下

saved_model_cli run --dir ./model_dir --tag_set serve --signature_def serving_default --input_exprs 'input=np.ones((20, 40, 3), dtype=np.float32)'
TF_Support

InvalidArgumentError通常是由输入中的数据类型不匹配引起的

根据您的错误,“ In [0]不是矩阵。而是具有形状[20,40,3] ”。
您可以尝试操纵输入数据以正确匹配模型最初训练的输入类型和形状。Python IDE相比,您还可以检查使用savedmodel_cli时模型如何处理您的输入当您使用Python IDE时可能会缺少一些预处理步骤,而这是在使用savedmodel_cli时完成的。

您可以在此链接中了解有关使用Saved_Model格式用法的更多信息

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章