如何在Keras中设置预测阈值？

Question

我用Keras训练了一个模型来做一个带有0和1标签的二进制分类任务，我在一系列图像上进行了测试。其中一些图像非常相似，预测为1，所以我想知道有没有办法设置一些阈值或分数来整理Keras中最有可能为1的图像？

Answer 1

如果您正在执行语义分段，请执行以下方法

执行

image = model.predict(your_input_image)
_,height,width,_ = image.shape()
image = image.reshape(height,width)
image[image >= threshold_percentage] = 255
image[image < threshold_percentage] = 0

如果是普通二进制文件

result = model.predict(your_input_image)
if result[0][1] > threshold_percentage:
    #belongs to class 1
else:
    #belongs to class 2