使用模型keras进行分割时预测图像的概率

时间:2020-10-23 21:36:17

标签: keras image-segmentation roc

所以我们正在建立一个细分(keras模型)。带有黑色和白色蒙版的灰度图像。 遮罩可以是全黑的,我们手动对其进行分类,然后将其分类为正,否则为负。

模型预测之后,我们看一下结果。 如果图像包含白色像素,我们将其手动分类为负像素,如果图像全黑则为正像素。所以我们得到了一个像[1,1,1,0,.....]

的numpy数组

我想知道是否有可能获得模型预测的每个像素或图像的概率。

要使用ROC曲线(sklearn.metrics.roc_curve)

我首先尝试在ROC曲线中输入Y_train(手动分类)和预测(也手动分类),但始终给出0.5

有什么想法吗?

0 个答案:

没有答案