ValueError:图层顺序的输入0与该图层不兼容:预期ndim = 4,找到的ndim = 3。收到的完整图形:[32、64、3]

时间:2019-11-24 18:45:51

标签: python tensorflow deep-learning tensorflow2.0

我绝对是深度学习的初学者,我正在尝试使用Tensorflow构建简单的分类模型。当然,我想在训练后使用此命令保存模型。

model.save('saved_model/my_model')

之后,我想在其他地方以及树莓派上使用我的模型。但是,我遇到了一些错误:

  

ValueError:图层顺序的输入0与   层:预期ndim = 4,找到的ndim = 3。收到完整形状:[32、64、3]

这是我的模型结构: my model

这是我的测试代码:

enter image description here

1 个答案:

答案 0 :(得分:0)

我的错,也许我研究不足。但是,我找到了解决方案。

在模型中,数据用255进行归一化。

...
img = img / 255.0
img = cv2.resize(img, (64, 64))
...