parent
6bc445f235
commit
37b236869f
@ -1,22 +0,0 @@
|
|||||||
"""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)
|
|
Loading…
Reference in new issue