You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
735 B
23 lines
735 B
8 years ago
|
"""Contains various CTC decoders in SWIG."""
|
||
|
|
from __future__ import absolute_import
|
||
|
|
from __future__ import division
|
||
|
|
from __future__ import print_function
|
||
|
|
|
||
|
|
from swig_ctc_beam_search_decoder import ctc_beam_search_decoder as beam_search_decoder
|
||
|
|
from swig_ctc_beam_search_decoder import ctc_best_path_decoder as best_path__decoder
|
||
|
|
|
||
|
|
|
||
|
|
def ctc_best_path_decoder(probs_seq, vocabulary):
|
||
|
|
best_path__decoder(probs_seq.to_list(), vocabulary)
|
||
|
|
|
||
|
|
|
||
|
|
def ctc_beam_search_decoder(
|
||
|
|
probs_seq,
|
||
|
|
beam_size,
|
||
|
|
vocabulary,
|
||
|
|
blank_id,
|
||
|
|
cutoff_prob=1.0,
|
||
|
|
ext_scoring_func=None, ):
|
||
|
|
beam_search_decoder(probs_seq.to_list(), beam_size, vocabulary, blank_id,
|
||
|
|
cutoff_prob, ext_scoring_func)
|