tf.keras.metrics.MeanIoU

Вычисляет метрику среднего пересечения над объединением.

Наследует от: IoU, Metric

tf.keras.metrics.MeanIoU(
    num_classes,
    name=None,
    dtype=None,
    ignore_class=None,
    sparse_y_true=True,
    sparse_y_pred=True,
    axis=-1
)

Формула:

iou = true_positives / (true_positives + false_positives + false_negatives)

Пересечение над объединением — это распространённая метрика оценки для семантической сегментации изображений.

Для вычисления IoU предсказания накапливаются в матрице ошибок с весами sample_weight, после чего на её основе вычисляется метрика.

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

Обратите внимание, что этот класс сначала вычисляет IoU для каждого отдельного класса, а затем возвращает среднее этих значений.

Аргументы
`num_classes`	Возможное количество меток в задаче предсказания. Это значение необходимо указать, поскольку будет выделена матрица ошибок размерности [num_classes, num_classes].
`name`	(Необязательно) Строковое имя экземпляра метрики.
`dtype`	(Необязательно) Тип данных результата метрики.
`ignore_class`	Необязательное целое число. Идентификатор класса, который следует игнорировать при вычислении метрики. Это полезно, например, в задачах сегментации, где на картах сегментации есть класс «void» (обычно -1 или 255). По умолчанию (`ignore_class=None`) учитываются все классы.
`sparse_y_true`	Указывает, закодированы ли метки целыми числами или плотными векторами с плавающей точкой. Если `False`, функция `argmax` используется для определения наиболее вероятной метки для каждого примера.
`sparse_y_pred`	Указывает, закодированы ли предсказания целыми числами или плотными векторами с плавающей точкой. Если `False`, функция `argmax` используется для определения наиболее вероятной метки для каждого примера.
`axis`	(Необязательно) Размерность, содержащая логиты. По умолчанию — `-1`.

Пример:

# cm = [[1, 1],
#        [1, 1]]
# sum_row = [2, 2], sum_col = [2, 2], true_positives = [1, 1]
# iou = true_positives / (sum_row + sum_col - true_positives))
# result = (1 / (2 + 2 - 1) + 1 / (2 + 2 - 1)) / 2 = 0.33
m = keras.metrics.MeanIoU(num_classes=2)
m.update_state([0, 0, 1, 1], [0, 1, 0, 1])
m.result()
0.33333334

m.reset_state()
m.update_state([0, 0, 1, 1], [0, 1, 0, 1],
               sample_weight=[0.3, 0.3, 0.3, 0.1])
m.result().numpy()
0.23809525

Использование с API compile():

model.compile(
    optimizer='sgd',
    loss='mse',
    metrics=[keras.metrics.MeanIoU(num_classes=2)])

Атрибуты
`dtype`
`variables`

Методы

`add_variable`

Просмотреть исходный код

add_variable(
    shape, initializer, dtype=None, aggregation='sum', name=None
)

`add_weight`

Просмотреть исходный код

add_weight(
    shape=(), initializer=None, dtype=None, name=None
)

`from_config`

Просмотреть исходный код

@classmethod
from_config(
    config
)

`get_config`

Просмотреть исходный код

get_config()

Возвращает сериализуемую конфигурацию метрики.

`reset_state`

Просмотреть исходный код

reset_state()

Сбрасывает все переменные состояния метрики.

Эта функция вызывается между эпохами/шагами, когда метрика оценивается во время обучения.

`result`

Просмотреть исходный код

result()

Вычисляет пересечение над объединением с помощью матрицы ошибок.

`stateless_reset_state`

Просмотреть исходный код

stateless_reset_state()

`stateless_result`

Просмотреть исходный код

stateless_result(
    metric_variables
)

`stateless_update_state`

Просмотреть исходный код

stateless_update_state(
    metric_variables, *args, **kwargs
)

`update_state`

Просмотреть исходный код

update_state(
    y_true, y_pred, sample_weight=None
)

Накапливает статистику матрицы ошибок.

Аргументы
`y_true`	Эталонные значения.
`y_pred`	Предсказанные значения.
`sample_weight`	Необязательные веса для каждого примера. Может быть `Tensor` с рангом 0 или равным рангу `y_true` и должен поддерживать широковещательное преобразование к `y_true`. По умолчанию — `1`.

Возвращаемое значение
Операция обновления.

`call`

Просмотреть исходный код

__call__(
    *args, **kwargs
)

Вызывает self как функцию.

© 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/keras/metrics/MeanIoU