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

问题描述

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

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

的numpy数组

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

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

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

有什么想法吗?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)