tf.metrics.mean_iou
Вычисление средней доли перекрытия (mIOU) на каждом шаге.
tf.metrics.mean_iou(
labels, predictions, num_classes, weights=None, metrics_collections=None,
updates_collections=None, name=None
)
Средняя доля перекрытия — распространённый показатель оценки семантической сегментации изображений, который сначала вычисляет долю перекрытия для каждого семантического класса, а затем вычисляет среднее значение по классам. Доля перекрытия определяется следующим образом: 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. |
© 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/metrics/mean_iou
