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