import gc
import psutil
import os
import matplotlib.pyplot as plt
import numpy as np

process = psutil.Process(os.getpid())
# initial memory usage
bytes_used_start = process.memory_info().rss

# generate 10 min data
fs = 20000
x = np.random.rand(fs*600)

for i in range(10):
    
    # create figure
    fig = plt.figure(figsize=(10,6))    
    ax = fig.add_axes([0, 0, 1, 1])   
    
    # plot spectrogram
    spectro, freqs, t, im = ax.specgram(
            x=x,
            NFFT=2**13,
            Fs=fs
            )
    
    ax.axes.set_ylim((9.8,10000))
    #ax.axes.set_yscale('log')      # -> comment this line in for memory leak
    
    fig.savefig('C:/Temp/spectro_{}.jpg'.format(i))
        
    # clean up
    plt.close(fig)
    gc.collect()
    
    # show memory usage
    bytes_used_cur =  process.memory_info().rss - bytes_used_start
    print('bytes used currently at step {} : {}'.format(i, bytes_used_cur))