Short and simple implementation of SpecAugment using numpy and functional programming.
Original SpecAugment Paper : https://doi.org/10.48550/arXiv.1904.08779
Installation : Clone the repository and run pip install . inside the project directory.
Example of usage
Librosa is required to run this example :
pip install librosa
importnumpyasnpimportlibrosafromSimpleSpecAugment.augmentimportspec_augmentsignal, sr=librosa.load(librosa.example('pistachio'), sr=16000)
spectrogram=librosa.feature.melspectrogram(y=signal, sr=sr, n_mels=128, fmax=8000)
#augment is a list containing the augmented spectrogramsaugment=spec_augment(spectrogram, augment_times=3, masks=1) 