back/models/importer_audio.py

import librosa.core.audio
import librosa.util
import numpy as np
from scipy.io import wavfile


def audio_importer(filename):
    # X : (n_samples, *dim, n_channels)
    "Generates data containing batch_size samples"

    # Generate data
    X = []
    # Load soundfile data
    data, r = librosa.core.audio.load(filename, sr=16384, mono=True, duration=3.0)
    n_samps = data.shape[0]
    print("N Samps in audio data: {}".format(n_samps))
    data = librosa.util.fix_length(data, size=1*16384)
    n_samps = data.shape[0]
    print("N Samps in audio data: {}".format(n_samps))
    X.append(data[: n_samps])
    Xd = np.expand_dims(np.vstack(X), axis=1)
    Xd = Xd.reshape((X.__len__(), 1, n_samps))

    return Xd

if __name__ == "__main__":
    audio_importer("C:/Users/yderre/Downloads/inver-synth-master/inver-synth-master/test_waves/InverSynth/InverSynth_00006.wav")