source: python/demos/demo_pitch_sinusoid.py @ 42dbcf4

feature/crepe
Last change on this file since 42dbcf4 was 1929be4, checked in by Ancor Cruz <hello@ancorcruz.com>, 7 years ago

python/demos/demo_pitch_sinusoid.py: use to yield an integer

  • Property mode set to 100755
File size: 1.7 KB
RevLine 
[7175ed4]1#! /usr/bin/env python
[e4c2169]2
[f6892d4]3import numpy as np
4import aubio
[e4c2169]5
6def build_sinusoid(length, freqs, samplerate):
[f6892d4]7    return np.sin( 2. * np.pi * np.arange(length) * freqs / samplerate).astype(aubio.float_type)
[e4c2169]8
9def run_pitch(p, input_vec):
[f6892d4]10    cands = []
11    for vec_slice in input_vec.reshape((-1, p.hop_size)):
12        a = p(vec_slice)[0]
13        cands.append(a)
14    return cands
[e4c2169]15
16methods = ['default', 'schmitt', 'fcomb', 'mcomb', 'yin', 'yinfft']
17
18cands = {}
19buf_size = 2048
20hop_size = 512
21samplerate = 44100
22sin_length = (samplerate * 10) % 512 * 512
[f6892d4]23freqs = np.zeros(sin_length)
[e4c2169]24
[4120fbc]25partition = sin_length // 8
[e4c2169]26pointer = 0
27
28pointer += partition
29freqs[pointer: pointer + partition] = 440
30
31pointer += partition
32pointer += partition
33freqs[ pointer : pointer + partition ] = 740
34
35pointer += partition
36freqs[ pointer : pointer + partition ] = 1480
37
38pointer += partition
39pointer += partition
[1929be4]40freqs[ pointer : pointer + partition ] = 400 + 5 * np.random.random(sin_length//8)
[e4c2169]41
42a = build_sinusoid(sin_length, freqs, samplerate)
43
44for method in methods:
[f6892d4]45    p = aubio.pitch(method, buf_size, hop_size, samplerate)
46    cands[method] = run_pitch(p, a)
[4120fbc]47    print(method)
48    print(cands[method])
[e4c2169]49
[4120fbc]50print("done computing")
[e4c2169]51
52if 1:
[f6892d4]53    import matplotlib.pyplot as plt
[e4c2169]54
[f6892d4]55    # times
56    ramp = np.arange(0, sin_length / hop_size).astype('float') * hop_size / samplerate
[e4c2169]57
[f6892d4]58    # plot each result
59    for method in methods:
60        plt.plot(ramp, cands[method], '.-', label=method)
[e4c2169]61
[f6892d4]62    # plot ground truth
63    ramp = np.arange(0, sin_length).astype('float') / samplerate
64    plt.plot(ramp, freqs, ':', label = 'ground truth')
65
66    plt.legend(loc='upper left')
67
68    plt.xlabel('time (s)')
69    plt.ylabel('frequency (Hz)')
70    plt.ylim([0,2000])
71    plt.show()
Note: See TracBrowser for help on using the repository browser.