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`	Возможная численность меток, которые может содержать задача прогнозирования. Это значение должно быть указано, так как будет выделена матрица неточностей размерности = [количество_классов, количество_классов].
`name`	(Необязательно) имя строки экземпляра метрики.
`dtype`	(Необязательно) тип данных результата метрики.
`ignore_class`	(Необязательно) целое число. Идентификатор класса, который нужно проигнорировать во время вычисления метрики. Это полезно, например, в задачах сегментации, имеющих класс «пустое значение» (обычно -1 или 255) в картах сегментации. По умолчанию (`ignore_class=None`) все классы учитываются.
`sparse_y_true`	Кодируются ли метки с использованием целых чисел или плотных векторов с плавающей точкой. Если `False`, используется функция `argmax`, чтобы определить наиболее вероятную метку каждого образца.
`sparse_y_pred`	Кодируются ли прогнозы с использованием целых чисел или плотных векторов с плавающей точкой. Если `False`, используется функция `argmax`, чтобы определить наиболее вероятную метку каждого образца.
`axis`	(Необязательно) измерение, содержащее значения logits. По умолчанию `-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