File size: 2,447 Bytes
1a7d583 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
import numpy as np
import matplotlib.pyplot as plt
import librosa.display
import librosa
def calculate_features(y, sr):
stft = np.abs(librosa.stft(y))
duration = librosa.get_duration(y=y, sr=sr)
cent = librosa.feature.spectral_centroid(S=stft, sr=sr)[0]
bw = librosa.feature.spectral_bandwidth(S=stft, sr=sr)[0]
rolloff = librosa.feature.spectral_rolloff(S=stft, sr=sr)[0]
return stft, duration, cent, bw, rolloff
def plot_title(title):
plt.suptitle(title, fontsize=16, fontweight="bold")
def plot_spectrogram(y, sr, stft, duration, cmap="inferno"):
plt.subplot(3, 1, 1)
plt.imshow(
librosa.amplitude_to_db(stft, ref=np.max),
origin="lower",
extent=[0, duration, 0, sr / 1000],
aspect="auto",
cmap=cmap, # Change the colormap here
)
plt.colorbar(format="%+2.0f dB")
plt.xlabel("Time (s)")
plt.ylabel("Frequency (kHz)")
plt.title("Spectrogram")
def plot_waveform(y, sr, duration):
plt.subplot(3, 1, 2)
librosa.display.waveshow(y, sr=sr)
plt.xlabel("Time (s)")
plt.ylabel("Amplitude")
plt.title("Waveform")
def plot_features(times, cent, bw, rolloff, duration):
plt.subplot(3, 1, 3)
plt.plot(times, cent, label="Spectral Centroid (kHz)", color="b")
plt.plot(times, bw, label="Spectral Bandwidth (kHz)", color="g")
plt.plot(times, rolloff, label="Spectral Rolloff (kHz)", color="r")
plt.xlabel("Time (s)")
plt.title("Spectral Features")
plt.legend()
def analyze_audio(audio_file, save_plot_path="logs/audio_analysis.png"):
y, sr = librosa.load(audio_file)
stft, duration, cent, bw, rolloff = calculate_features(y, sr)
plt.figure(figsize=(12, 10))
plot_title("Audio Analysis" + " - " + audio_file.split("/")[-1])
plot_spectrogram(y, sr, stft, duration)
plot_waveform(y, sr, duration)
plot_features(librosa.times_like(cent), cent, bw, rolloff, duration)
plt.tight_layout()
if save_plot_path:
plt.savefig(save_plot_path, bbox_inches="tight", dpi=300)
plt.close()
audio_info = f"""Sample Rate: {sr}\nDuration: {(
str(round(duration, 2)) + " seconds"
if duration < 60
else str(round(duration / 60, 2)) + " minutes"
)}\nNumber of Samples: {len(y)}\nBits per Sample: {librosa.get_samplerate(audio_file)}\nChannels: {"Mono (1)" if y.ndim == 1 else "Stereo (2)"}"""
return audio_info, save_plot_path
|