Spaces:
Runtime error
Runtime error
Update helper.py
Browse files
helper.py
CHANGED
|
@@ -1,3 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
def plot_spectrogram(stft, title="Spectrogram", xlim=None):
|
| 2 |
magnitude = stft.abs()
|
| 3 |
spectrogram = 20 * torch.log10(magnitude + 1e-8).numpy()
|
|
|
|
| 1 |
+
import torch
|
| 2 |
+
import matplotlib.pyplot as plt
|
| 3 |
+
from pesq import pesq
|
| 4 |
+
from pystoi import stoi
|
| 5 |
+
import mir_eval
|
| 6 |
+
REFERENCE_CHANNEL = 0
|
| 7 |
def plot_spectrogram(stft, title="Spectrogram", xlim=None):
|
| 8 |
magnitude = stft.abs()
|
| 9 |
spectrogram = 20 * torch.log10(magnitude + 1e-8).numpy()
|