To answer my own question for anyone else struggling:
im = ax.images
cb = im[-1].colorbar
cb.set_label('Amplitude [dB]')
To answer my own question for anyone else struggling:
im = ax.images
cb = im[-1].colorbar
cb.set_label('Amplitude [dB]')