In [ ]:

    

import matplotlib.pyplot as plt
import numpy as np
from PIL import Image
from scipy import signal
from scipy.io import wavfile


    

In [ ]:

    

prefix = 'gs://path/to/audio/files'
spectro_prefix = 'gs://path/to/spectrograms'

for i in range(len(names)):
    fname = prefix + names[i]
    !gsutil cp $fname .

    # Create spectrogram with scipy
    sample_rate, samples = wavfile.read(names[i])
    freq, times, spectro = signal.spectrogram(samples, sample_rate)
    
    # Create specrogram image with matplotlib
    plt.figure(figsize=(28,28), frameon=False)
    fig = plt.gcf()
    plt.axis('off')
    plt.pcolormesh(times, freq, np.log(spectro))

    # Save spectrogram to local file
    name = names[i].split('.')[0]
    fig.savefig(name)
    spectro_path = spectro_prefix + name + '.png'
    local_path = name + '.png'

    # Convert image to 3 channels and shrink
    im = Image.open(local_path)
    im.thumbnail((128,128), Image.ANTIALIAS)
    im.convert(mode='RGB')
    im.save(local_path)

    # Save spectrogram to GCS and remove local files
    !gsutil cp $local_path $spectro_path
    !rm $local_path
    local_audio = names[i]
    !rm $local_audio