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

#python #tensorflow #keras #tf.keras #semantic-segmentation

Вопрос:

Я обучил UNET и рассчитал среднее значение с помощью [tf.keras.metrics.MeanIoU(num_classes=2)] . Я получил среднее значение=45% при оценке на тестовом наборе. Используя y_pred = model.predict(Test_Images) , я могу получать прогнозы. Как мне получить значения IOU для каждого тестового изображения?