Spec-Zone.ru › TensorFlow 2.9

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 = истинные_положительные / (истинные_положительные + ложные_положительные + ложные_отрицательные). Предсказания накапливаются в матрице неточностей, взвешенной weights, а затем по ней вычисляется mIOU.

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

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

Аргументы
labels A Tensor меток истинного значения с формой [размер пакета] и типа int32 или int64. Tensor будет сжат, если его ранг > 1.
predictions A Tensor результатов предсказания для семантических меток, форма которого [размер пакета] и тип int32 или int64. Tensor будет сжат, если его ранг > 1.
num_classes Возможное количество меток, которые может иметь задача прогнозирования. Это значение должно быть предоставлено, так как будет выделена матрица неточностей размерности = [число_классов, число_классов].
weights Необязательные Tensor, ранг которых равен 0 или такому же, как у labels, и должен быть совместим с labels (т. е. все измерения должны быть либо 1, либо совпадать с соответствующим измерением labels).
metrics_collections Необязательный список коллекций, в которые mean_iou следует добавить.
updates_collections Необязательный список коллекций, в которые update_op следует добавить.
name Необязательное имя области переменных.
Возвращает
mean_iou A 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/versions/r2.9/api_docs/python/tf/compat/v1/metrics/mean_iou

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API