我绝对是深度学习的初学者,我正在尝试使用Tensorflow构建简单的分类模型。当然,我想在训练后使用此命令保存模型。
model.save('saved_model/my_model')
之后,我想在其他地方以及树莓派上使用我的模型。但是,我遇到了一些错误:
ValueError:图层顺序的输入0与 层:预期ndim = 4,找到的ndim = 3。收到完整形状:[32、64、3]
这是我的测试代码:
答案 0 :(得分:0)
我的错,也许我研究不足。但是,我找到了解决方案。
在模型中,数据用255进行归一化。
...
img = img / 255.0
img = cv2.resize(img, (64, 64))
...