抓张量流预测精度

Question

model = tf.keras.models.load_model(".\\models\\dogandcat")
prediction = model.predict([prepare(pic)])  
if CATEGORIES[int(prediction[0][0])] == 'Dog':
        print('Dog')

这是我当前获取图像类型的代码。一旦检测到狗，我将尝试使它查看检测的准确性/准确性。希望这有意义吗？

Answer 1

如果通过

我正在尝试使其检测的准确性/准确性...

您的意思是模型为“狗”类型提供给图像的概率，那么您应该仅获取logits并查看狗的索引（可以从CATEGORIES变量推断出）。

logits 是模型的最后一个softmax层的输出。