Tensorflow为每一行预测相同的值

数据挖掘 深度学习 张量流 预言
2022-02-02 00:21:40

我有一个训练有素的模型。对于单个预测,我恢复最后一个检查点并传递单个图像进行预测,但每一行的结果都是相同的。有谁知道可能出了什么问题?

这是预测的代码:

在此处输入图像描述

这是输出:

在此处输入图像描述

1个回答

我想我找到了问题所在。我需要输入的形状为 (1, 256, 256, 1),这就是为什么我执行 image = image[...,0][...,None]/255 但这样做将所有值更改为 0