Make variables consistent
This commit is contained in:
parent
d9a35d63b0
commit
9a92fa40ca
|
@ -900,21 +900,21 @@ def do_single_file_inference(input_file_path):
|
||||||
features = create_overlapping_windows(features).eval(session=session)
|
features = create_overlapping_windows(features).eval(session=session)
|
||||||
features_len = features_len.eval(session=session)
|
features_len = features_len.eval(session=session)
|
||||||
|
|
||||||
logits = outputs['outputs'].eval(feed_dict={
|
probs = outputs['outputs'].eval(feed_dict={
|
||||||
inputs['input']: features,
|
inputs['input']: features,
|
||||||
inputs['input_lengths']: features_len,
|
inputs['input_lengths']: features_len,
|
||||||
inputs['previous_state_c']: previous_state_c,
|
inputs['previous_state_c']: previous_state_c,
|
||||||
inputs['previous_state_h']: previous_state_h,
|
inputs['previous_state_h']: previous_state_h,
|
||||||
}, session=session)
|
}, session=session)
|
||||||
|
|
||||||
logits = np.squeeze(logits)
|
probs = np.squeeze(probs)
|
||||||
|
|
||||||
if FLAGS.scorer_path:
|
if FLAGS.scorer_path:
|
||||||
scorer = Scorer(FLAGS.lm_alpha, FLAGS.lm_beta,
|
scorer = Scorer(FLAGS.lm_alpha, FLAGS.lm_beta,
|
||||||
FLAGS.scorer_path, Config.alphabet)
|
FLAGS.scorer_path, Config.alphabet)
|
||||||
else:
|
else:
|
||||||
scorer = None
|
scorer = None
|
||||||
decoded = ctc_beam_search_decoder(logits, Config.alphabet, FLAGS.beam_width,
|
decoded = ctc_beam_search_decoder(probs, Config.alphabet, FLAGS.beam_width,
|
||||||
scorer=scorer, cutoff_prob=FLAGS.cutoff_prob,
|
scorer=scorer, cutoff_prob=FLAGS.cutoff_prob,
|
||||||
cutoff_top_n=FLAGS.cutoff_top_n)
|
cutoff_top_n=FLAGS.cutoff_top_n)
|
||||||
# Print highest probability result
|
# Print highest probability result
|
||||||
|
|
Loading…
Reference in New Issue