tf.keras.metrics.IoU

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

Наследуется от: Metric

tf.keras.metrics.IoU(
    num_classes,
    target_class_ids,
    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 для каждого отдельного класса, а затем возвращает среднее значение IoU для классов, указанных в target_class_ids. Если target_class_ids содержит только один идентификатор, возвращается IoU именно этого класса.

Аргументы
`num_classes`	Возможное количество меток в задаче предсказания.
`target_class_ids`	Кортеж или список идентификаторов целевых классов, для которых возвращается метрика. Чтобы вычислить IoU для конкретного класса, необходимо указать список (или кортеж) с одним идентификатором.
`name`	(Необязательно) строковое имя экземпляра метрики.
`dtype`	(Необязательно) тип данных результата метрики.
`ignore_class`	Необязательное целое число. Идентификатор класса, который следует игнорировать при вычислении метрики. Это полезно, например, в задачах сегментации с классом «void» (обычно -1 или 255) на картах сегментации. По умолчанию (`ignore_class=None`) учитываются все классы.
`sparse_y_true`	Указывает, закодированы ли метки целыми числами или плотными векторами с плавающей точкой. Если `False`, функция `argmax` используется для определения наиболее вероятной метки для каждого образца.
`sparse_y_pred`	Указывает, закодированы ли предсказания целыми числами или плотными векторами с плавающей точкой. Если `False`, функция `argmax` используется для определения наиболее вероятной метки для каждого образца.
`axis`	(Необязательно) -1 — размерность, содержащая логиты. По умолчанию — `-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))
# iou = [0.33, 0.33]
m = keras.metrics.IoU(num_classes=2, target_class_ids=[0])
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])
# cm = [[0.3, 0.3],
#        [0.3, 0.1]]
# sum_row = [0.6, 0.4], sum_col = [0.6, 0.4],
# true_positives = [0.3, 0.1]
# iou = [0.33, 0.14]
m.result()
0.33333334

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

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

Атрибуты
`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/IoU