data.py 文件源码

def transform_wav_data_op(wav_data_tensor, hparams, is_training,
                          jitter_amount_sec):
  """Transforms wav data."""
  def transform_wav_data(wav_data):
    """Transforms wav data."""
    # Only do audio transformations during training.
    if is_training:
      wav_data = audio_io.jitter_wav_data(wav_data, hparams.sample_rate,
                                          jitter_amount_sec)

    # Normalize.
    if hparams.normalize_audio:
      wav_data = audio_io.normalize_wav_data(wav_data, hparams.sample_rate)

    return [wav_data]

  return tf.py_func(
      transform_wav_data,
      [wav_data_tensor],
      tf.string,
      name='transform_wav_data_op')