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
)

Среднее пересечение по объединению — распространённая метрика оценки семантической сегментации изображений. Сначала вычисляется IOU для каждого семантического класса, а затем рассчитывается среднее значение по классам. IOU определяется следующим образом: IOU = true_positive / (true_positive + false_positive + false_negative). Предсказания накапливаются в матрице ошибок с весами weights, после чего по ней вычисляется mIOU.

Для оценки метрики на потоке данных функция создаёт операцию update_op, которая обновляет эти переменные и возвращает mean_iou.

Если weights имеет значение None, веса по умолчанию равны 1. Используйте веса, равные 0, чтобы маскировать значения.

Аргументы
`labels`	`Tensor` меток истинных значений с формой [размер пакета] и типом `int32` или `int64`. Если ранг тензора > 1, он будет преобразован в одномерный.
`predictions`	`Tensor` результатов предсказания семантических меток с формой [размер пакета] и типом `int32` или `int64`. Если ранг тензора > 1, он будет преобразован в одномерный.
`num_classes`	Возможное количество меток в задаче предсказания. Это значение необходимо указать, поскольку будет выделена матрица ошибок размерности [num_classes, num_classes].
`weights`	Необязательный `Tensor` с рангом 0 или тем же рангом, что и у `labels`; он должен допускать широковещательное преобразование к `labels` (то есть все размерности должны быть либо `1`, либо совпадать с соответствующими размерностями `labels`).
`metrics_collections`	Необязательный список коллекций, в которые следует добавить `mean_iou`.
`updates_collections`	Необязательный список коллекций, в которые следует добавить `update_op`.
`name`	Необязательное имя variable_scope.

Возвращает
`mean_iou`	`Tensor`, представляющий среднее пересечение по объединению.
`update_op`	Операция, увеличивающая значения в матрице ошибок.

Вызывает исключения
`ValueError`	Если формы `predictions` и `labels` не совпадают, если `weights` не является `None` и его форма не совпадает с формой `predictions` или если `metrics_collections` либо `updates_collections` не является списком или кортежем.
`RuntimeError`	Если включён режим eager execution.

© 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