def main(*args, **kwargs):
FLAGS(sys.argv, known_only=True)
if FLAGS.enable_eager_execution:
tf.enable_eager_execution()
else:
tf.disable_eager_execution()
FLAGS.unparse_flags()
tf.test.main(*args, **kwargs)
f = tar.extractfile(tarinfo)
wav_bytes = audio_lib.DecodeFlacToWav(f.read())
f.close()
frames = sess.run(log_mel, feed_dict={tf_bytes: wav_bytes})
assert uttid in trans, uttid
num_words = len(trans[uttid])
tf.logging.info('utt[%d]: %s [%d frames, %d words]', n, uttid,
frames.shape[1], num_words)
ex = _MakeTfExample(uttid, frames, trans[uttid])
outf = _SelectRandomShard(recordio_writers)
outf.write(ex.SerializeToString())
tar.close()
file_obj.close()
_CloseSubShards(recordio_writers)
def main(_):
tf.logging.set_verbosity(tf.logging.INFO)
if FLAGS.dump_transcripts:
_DumpTranscripts()
elif FLAGS.generate_tfrecords:
_CreateAsrFeatures()
else:
tf.logging.error(
'Nothing to do! Use --dump_transcripts or --generate_tfrecords')
if __name__ == '__main__':
tf.disable_eager_execution()
tf.app.run(main)
