tf.keras.metrics.BinaryIoU
Вычисляет метрику пересечения над объединением для класса 0 и/или 1.
tf.keras.metrics.BinaryIoU(
target_class_ids=(0, 1), threshold=0.5, name=None, dtype=None
)
Формула:
iou = true_positives / (true_positives + false_positives + false_negatives)
Пересечение над объединением — это распространённая метрика оценки семантической сегментации изображений.
Для вычисления IoU предсказания накапливаются в матрице ошибок с учётом весов sample_weight, после чего по ней вычисляется метрика.
Если sample_weight — None, вес по умолчанию равен 1. Используйте sample_weight со значением 0, чтобы маскировать значения.
Этот класс можно использовать для вычисления IoU в задаче бинарной классификации, в которой предсказания представлены логитами. Сначала к предсказанным значениям применяется threshold: значения ниже threshold преобразуются в класс 0, а значения выше threshold — в класс 1.
Затем вычисляется IoU для классов 0 и 1, а возвращается среднее значение IoU для классов, указанных с помощью target_class_ids.
Примечание: при использованииthreshold=0эта метрика ведёт себя так же, какIoU.
| Аргументы | |
|---|---|
target_class_ids | Кортеж или список идентификаторов целевых классов, для которых возвращается метрика. Возможные значения: [0], [1] или [0, 1]. При значении [0] (или [1]) возвращается метрика IoU для класса 0 (или, соответственно, класса 1). При значении [0, 1] возвращается среднее значение IoU для двух классов. |
threshold | Пороговое значение, применяемое к логитам предсказаний для преобразования в предсказанный класс 0, если логит меньше threshold, или в предсказанный класс 1, если логит больше threshold. |
name | (Необязательно) строковое имя экземпляра метрики. |
dtype | (Необязательно) тип данных результата метрики. |
Пример:
Пример:
m = keras.metrics.BinaryIoU(target_class_ids=[0, 1], threshold=0.3) m.update_state([0, 1, 0, 1], [0.1, 0.2, 0.4, 0.7]) m.result() 0.33333334
m.reset_state()
m.update_state([0, 1, 0, 1], [0.1, 0.2, 0.4, 0.7],
sample_weight=[0.2, 0.3, 0.4, 0.1])
# cm = [[0.2, 0.4],
# [0.3, 0.1]]
# sum_row = [0.6, 0.4], sum_col = [0.5, 0.5],
# true_positives = [0.2, 0.1]
# iou = [0.222, 0.125]
m.result()
0.17361112Использование с API compile():
model.compile(
optimizer='sgd',
loss='mse',
metrics=[keras.metrics.BinaryIoU(
target_class_ids=[0],
threshold=0.5
)]
)
| Атрибуты | |
|---|---|
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
)
Накапливает статистику матрицы ошибок.
Перед обновлением матрицы ошибок для предсказанных значений применяется пороговое значение: 0 для значений меньше threshold, 1 для значений, больших или равных threshold
| Аргументы | |
|---|---|
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/BinaryIoU