tf.compat.v1.metrics.mean_iou
Вычисление средней доли пересечения над объединением (mIOU) на каждом шаге.
tf.compat.v1.metrics.mean_iou(
labels,
predictions,
num_classes,
weights=None,
metrics_collections=None,
updates_collections=None,
name=None
)
Средняя доля пересечения над объединением (mIOU) — это распространенная метрика оценки для семантической сегментации изображений, которая сначала вычисляет IOU для каждого семантического класса, а затем вычисляет среднее значение по классам. IOU определяется следующим образом: IOU = true_positive / (true_positive + false_positive + false_negative). Предсказания накапливаются в матрице неточностей, взвешенной значением weights, а затем по ней вычисляется mIOU.
Для оценки метрики по потоку данных функция создаёт операцию update_op, которая обновляет эти переменные и возвращает mean_iou.
Если значение weights равно None, веса по умолчанию равны 1. Используйте веса 0 для маскирования значений.
| Аргументы | |
|---|---|
labels | Массив меток истинных значений с формой [размер пакета] и типом int32 или int64. Массив будет сглажен, если его ранг > 1. |
predictions | Массив результатов предсказания для семантических меток, форма которого [размер пакета] и тип int32 или int64. Массив будет сглажен, если его ранг > 1. |
num_classes | Возможное количество меток, которые может иметь задача предсказания. Это значение необходимо предоставить, поскольку будет выделена матрица неточностей размерностью = [число_классов, число_классов]. |
weights | Необязательные веса, ранг которых равен 0 или совпадает с рангом labels, и которые должны быть совместимы для трансляции с labels (т.е., все измерения должны быть либо 1, либо такими же, как соответствующее измерение labels). |
metrics_collections | Необязательный список коллекций, в которые необходимо добавить mean_iou. |
updates_collections | Необязательный список коллекций, в которые необходимо добавить update_op. |
name | Необязательное имя области переменных. |
| Возвращаемое значение | |
|---|---|
mean_iou | Tensor, представляющий среднюю долю пересечения над объединением. |
update_op | Операция, увеличивающая матрицу неточностей. |
| Исключения | |
|---|---|
ValueError | Если формы predictions и labels несовместимы, или если weights не является None и его форма не соответствует predictions, или если metrics_collections или updates_collections не являются списком или кортежем. |
RuntimeError | Если включено выполнение Eager. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/compat/v1/metrics/mean_iou