Как получить значение вероятности «обнаруженного» объекта?

Question

inp - изображение.

 prediction = model.predict([inp])

Я тренировал его с кошками и собаками соответственно.

    if int(prediction[0][0]) == 0:
        print("it is a cat")

Как узнать вероятности появления кошки или собаки для этого ввода?

Answer 1

Тогда у вас есть различие между собакой и кошкой. Понимание того, что вы пометили данные 0 как кошку, а 1 как собаку.

Выходные данные представляют двоичное событие, когда значение, близкое к нулю, равно коту, а когда значение близко к 1 - это собака. p (собака) = предсказание и p (кошка) = 1 - p (собака).

Исследование, называемое ROC-кривой или PR-кривой , обычно проводится для определения того, в какой момент проводить различие между кошкой и собакой.

Но в вашем случае, чтобы идти быстро, вы можете рассмотреть 0,5, то есть:

p = model.predict([inp]) 
is_cat = p <= 0.5
is_dog = p> 0.5