tf.keras.ops.ctc_decode

Декодирует выходные данные модели CTC.

Просмотреть псевдонимы

Основные псевдонимы

tf.keras.ops.nn.ctc_decode

tf.keras.ops.ctc_decode(
    inputs,
    sequence_lengths,
    strategy='greedy',
    beam_width=100,
    top_paths=1,
    merge_repeated=True,
    mask_index=None
)

Аргументы
`inputs`	Тензор формы `(batch_size, max_length, num_classes)`, содержащий логиты (выходные данные модели). Их не следует нормализовать с помощью softmax.
`sequence_lengths`	Тензор формы `(batch_size,)`, содержащий длины последовательностей в пакете.
`strategy`	Строка, задающая стратегию декодирования. Поддерживаемые значения: `"greedy"` и `"beam_search"`.
`beam_width`	Целочисленный скаляр, задающий ширину луча при поиске по лучу. Значение по умолчанию — 100.
`top_paths`	Целочисленный скаляр, задающий количество возвращаемых лучших путей. Значение по умолчанию — 1.
`merge_repeated`	Логический скаляр, указывающий, следует ли объединять повторяющиеся метки в выходных данных. Значение по умолчанию — `True`.
`mask_index`	Целочисленный скаляр, задающий индекс символа маски в словаре. Значение по умолчанию — `None`.

Возвращает

Возвращает
Кортеж, содержащий: Тензор, представляющий список декодированных последовательностей. Если `strategy="greedy"`, его форма — `(1, batch_size, max_length)`. Если `strategy="beam_seatch"`, его форма — `(top_paths, batch_size, max_length)`. Обратите внимание: `-1` обозначает пустую метку. Если `strategy="greedy"`, тензор формы `(batch_size, 1)`, представляющий отрицательную сумму логитов вероятностей для каждой последовательности. Если `strategy="beam_seatch"`, тензор формы `(batch_size, top_paths)`, представляющий логарифм вероятности для каждой последовательности.

Кортеж, содержащий:

Тензор, представляющий список декодированных последовательностей. Если strategy="greedy", его форма — (1, batch_size, max_length). Если strategy="beam_seatch", его форма — (top_paths, batch_size, max_length). Обратите внимание: -1 обозначает пустую метку.
Если strategy="greedy", тензор формы (batch_size, 1), представляющий отрицательную сумму логитов вероятностей для каждой последовательности. Если strategy="beam_seatch", тензор формы (batch_size, top_paths), представляющий логарифм вероятности для каждой последовательности.

© 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/ops/ctc_decode