spectrogram_generator.py

# -*- coding: utf-8 -*-
"""
@author: Ankit

This script creates spectrograms of corresponding wav audio files

"""

import matplotlib.pyplot as plt
from os import listdir
from os.path import isfile,join
import os
import librosa
import librosa.display
import librosa
import librosa.display
import numpy as np

path="C:\\Users\\Ankit\\Desktop\\Scalable_Audio_Captcha\\audio_captcha_wav\\"
dest = "C:\\Users\\Ankit\\Desktop\\Scalable_Audio_Captcha\\audio_captcha_spectrogram\\"
files = [f for f in listdir(path) if isfile(join(path,f))]
for audio in files:
    x, sr = librosa.load(os.path.join(path, audio))
    plt.figure(figsize=(1.28, .64), dpi=100)
    plt.axis('off')
    plt.axes([0., 0., 1., 1., ], frameon=False, xticks=[], yticks=[])
    S = librosa.feature.melspectrogram(y=x, sr=sr)
    librosa.display.specshow(librosa.power_to_db(S, ref=np.max))
    plt.savefig(os.path.join(dest, os.path.splitext(audio)[0]+'.png'), bbox_inches=None, pad_inches=0)
    plt.close()