experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add missing imports for sample rate normalization

Browse Source
This commit is contained in:
Reuben Morais 2021-04-06 12:45:32 +02:00
parent 39627b282c
commit 449a723bf6
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
training/coqui_stt_training/evaluate.py
View File

@ -14,7 +14,7 @@ import tensorflow.compat.v1 as tfv1
from ds_ctcdecoder import ctc_beam_search_decoder_batch, Scorer from ds_ctcdecoder import ctc_beam_search_decoder_batch, Scorer
from six.moves import zip from six.moves import zip
from .util.augmentations import NormalizeSampleRate
from .util.config import Config, initialize_globals from .util.config import Config, initialize_globals
from .util.checkpoints import load_graph_for_evaluation from .util.checkpoints import load_graph_for_evaluation
from .util.evaluate_tools import calculate_and_print_report, save_samples_json from .util.evaluate_tools import calculate_and_print_report, save_samples_json
@ -53,6 +53,7 @@ def evaluate(test_csvs, create_model):
test_sets = [create_dataset([csv], test_sets = [create_dataset([csv],
batch_size=FLAGS.test_batch_size, batch_size=FLAGS.test_batch_size,
train_phase=False, train_phase=False,
augmentations=[NormalizeSampleRate(FLAGS.audio_sample_rate)],
reverse=FLAGS.reverse_test, reverse=FLAGS.reverse_test,
limit=FLAGS.limit_test) for csv in test_csvs] limit=FLAGS.limit_test) for csv in test_csvs]
iterator = tfv1.data.Iterator.from_structure(tfv1.data.get_output_types(test_sets[0]), iterator = tfv1.data.Iterator.from_structure(tfv1.data.get_output_types(test_sets[0]),

1
training/coqui_stt_training/train.py
View File

@ -28,6 +28,7 @@ from datetime import datetime
from ds_ctcdecoder import ctc_beam_search_decoder, Scorer from ds_ctcdecoder import ctc_beam_search_decoder, Scorer
from .evaluate import evaluate from .evaluate import evaluate
from six.moves import zip, range from six.moves import zip, range
from .util.augmentations import NormalizeSampleRate
from .util.config import Config, initialize_globals from .util.config import Config, initialize_globals
from .util.checkpoints import load_or_init_graph_for_training, load_graph_for_evaluation, reload_best_checkpoint from .util.checkpoints import load_or_init_graph_for_training, load_graph_for_evaluation, reload_best_checkpoint
from .util.evaluate_tools import save_samples_json from .util.evaluate_tools import save_samples_json