我使用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 正好是 1224224*3 所以位图的大小应该没问题 . 但是为什么Tensor的形状1,224,224,1和我怎么能改变呢?