使用.pb文件时输入Tensor错误

时间:2018-02-02 08:09:44

标签: android tensorflow keras protocol-buffers

我使用Keras和TensorFlow将MobileNet模型转换为.pb文件,我想在Android中使用它。序列化显然工作得很好,当我在TensorBoard中可视化.pb文件时,它显示了4D输入形状(?,?,?,3),我想这很好,因为我想以格式提供图像(1,224,224,3)。 但是在加载模型后,将所需的位图转换为该格式并通过TensorFlowInferenceInterface将其提供给TensorFlow,我收到以下错误:

"IllegalArgumentException: buffer with 150528 elements is not compatible with a Tensor with shape [1, 224, 224, 1]"

150528 正好是 1 * 224 * 224 * 3 所以位图的大小应该没问题。但是为什么Tensor的形状1,224,224,1和我怎么能改变呢?

0 个答案:

没有答案