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

Question

I'm trained a model with Keras to do a binary classification task with 0 and 1 labels, and I tested it on a series of images. Some of these images are very similar and were predicted as 1, so I want to know is there a way to set some threshold or scores to sort out the images that most likely to be 1 in Keras?

Answer 1

if you are performing semantic segmentation then carry out the below method

perform

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

if normal binary

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