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