Creates a spectrogram based on a waveform

build_spectrogram.py

import tensorflow as tf
import tensorflow_io as tfio

tensor = tf.cast(audio_tensor, tf.float32) / 32768.0

spectrogram = tfio.audio.spectrogram(
    tensor, nfft=512, window=512, stride=256)

mel_spectrogram= tfio.audio.melscale(
    spectrogram, rate=16000, mels=128, fmin=0, fmax=8000)

dbscale_mel_spectrogram= tfio.audio.dbscale(mel_spectrogram, top_db=80)