我使用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和我怎么能改变呢?