tf.contrib.metrics.streaming_mean_iou
Вычисление среднего значения пересечения-над-объединением (mIOU) на каждом шаге.
tf.contrib.metrics.streaming_mean_iou(
predictions, labels, 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 для маскирования значений.
| Аргументы | |
|---|---|
predictions | Результат предсказаний семантических меток в виде Tensor, размер которого равен [размер пакета], а тип — int32 или int64. Тензор будет уплощён, если его ранг > 1. |
labels | Результат истинных меток в виде Tensor, размер которого равен [размер пакета], а тип — int32 или int64. Тензор будет уплощён, если его ранг > 1. |
num_classes | Возможное количество меток, которые может иметь задача предсказания. Это значение должно быть предоставлено, так как будет выделена матрица неточностей размерности = [число классов, число классов]. |
weights | Необязательный Tensor с формой, совместимой с predictions. |
metrics_collections | Необязательный список коллекций, к которым mean_iou должен быть добавлен. |
updates_collections | Необязательный список коллекций, к которым update_op должен быть добавлен. |
name | Необязательное имя области переменных. |
| Возвращает | |
|---|---|
mean_iou | Tensor представляющий среднее значение пересечения-над-объединением. |
update_op | Операция, увеличивающая матрицу неточностей. |
| Возбуждает | |
|---|---|
ValueError | Если predictions и labels имеют несовместимые формы, или если weights не None и его форма не соответствует predictions, или если metrics_collections или updates_collections не являются списком или кортежем. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/metrics/streaming_mean_iou
