File size: 1,469 Bytes
2bbfbb7 0393dfa 2bbfbb7 0393dfa 2bbfbb7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
//! Benchmark for mel-spectrogram computation
use criterion::{black_box, criterion_group, criterion_main, Criterion};
use indextts::audio::{mel_spectrogram, AudioConfig};
fn bench_mel_spectrogram(c: &mut Criterion) {
let config = AudioConfig::default();
// Generate 1 second of audio
let num_samples = config.sample_rate as usize;
let signal: Vec<f32> = (0..num_samples).map(|i| (i as f32 * 0.01).sin()).collect();
c.bench_function("mel_spectrogram_1s", |b| {
b.iter(|| mel_spectrogram(black_box(&signal), black_box(&config)))
});
// Generate 10 seconds of audio
let long_signal: Vec<f32> = (0..num_samples * 10)
.map(|i| (i as f32 * 0.01).sin())
.collect();
c.bench_function("mel_spectrogram_10s", |b| {
b.iter(|| mel_spectrogram(black_box(&long_signal), black_box(&config)))
});
}
fn bench_stft(c: &mut Criterion) {
let config = AudioConfig::default();
let num_samples = config.sample_rate as usize;
let signal: Vec<f32> = (0..num_samples).map(|i| (i as f32 * 0.01).sin()).collect();
c.bench_function("stft_1s", |b| {
b.iter(|| {
indextts::audio::mel::stft(
black_box(&signal),
black_box(config.n_fft),
black_box(config.hop_length),
black_box(config.win_length),
)
})
});
}
criterion_group!(benches, bench_mel_spectrogram, bench_stft);
criterion_main!(benches);
|