view gmm_baseline_experiments/external_libs/librosa/librosa-0.3.1/examples/adjust_tuning.py @ 5:b523456082ca tip

Update path to dataset and reflect modified chunk naming convention.

#!/usr/bin/env python
'''CREATED:2014-05-22 16:43:44 by Brian McFee <[e-mail]>

Pitch-shift a recording to be in A440 tuning.

Usage: ./adjust_tuning.py [-h] input_file output_file
'''
from __future__ import print_function

import argparse
import sys
import librosa


def adjust_tuning(input_file, output_file):
    '''Load audio, estimate tuning, apply pitch correction, and save.'''
    print('Loading ', input_file)
    y, sr = librosa.load(input_file)

    print('Separating harmonic component ... ')
    y_harm = librosa.effects.harmonic(y)

    print('Estimating tuning ... ')
    # Just track the pitches associated with high magnitude
    tuning = librosa.feature.estimate_tuning(y=y_harm, sr=sr)

    print('{:+0.2f} cents'.format(100 * tuning))
    print('Applying pitch-correction of {:+0.2f} cents'.format(-100 * tuning))
    y_tuned = librosa.effects.pitch_shift(y, sr, -tuning)

    print('Saving tuned audio to: ', output_file)
    librosa.output.write_wav(output_file, y_tuned, sr)


def process_arguments(args):
    '''Argparse function to get the program parameters'''

    parser = argparse.ArgumentParser(description='Tuning adjustment example')

    parser.add_argument('input_file',
                        action='store',
                        help='path to the input file (wav, mp3, etc)')

    parser.add_argument('output_file',
                        action='store',
                        help='path to store the output wav')

    return vars(parser.parse_args(args))


if __name__ == '__main__':
    # Get the parameters
    params = process_arguments(sys.argv[1:])

    # Run the beat tracker
    adjust_tuning(params['input_file'], params['output_file'])