Update app.py

app.py

@@ -49,13 +49,13 @@ def visualize(hr, lr, recon):
     window = np.hanning(window_size)
 
     stft_hr = librosa.core.spectrum.stft(hr, n_fft=window_size, hop_length=512, window=window)
-    stft_hr = 2 * np.abs(stft_hr) / np.sum(window)
+    #stft_hr = 2 * np.abs(stft_hr) / np.sum(window)
 
     stft_lr = librosa.core.spectrum.stft(lr, n_fft=window_size, hop_length=512, window=window)
-    stft_lr = 2 * np.abs(stft_lr) / np.sum(window)
+    #stft_lr = 2 * np.abs(stft_lr) / np.sum(window)
 
     stft_recon = librosa.core.spectrum.stft(recon, n_fft=window_size, hop_length=512, window=window)
-    stft_recon = 2 * np.abs(stft_recon) / np.sum(window)
+    #stft_recon = 2 * np.abs(stft_recon) / np.sum(window)
 
     fig, (ax1, ax2, ax3) = plt.subplots(3, 1, sharey=True, sharex=True, figsize=(16, 12))
     ax1.title.set_text('Оригинальный сигнал')
@@ -63,9 +63,9 @@ def visualize(hr, lr, recon):
     ax3.title.set_text('Улучшенный сигнал')
 
     canvas = FigureCanvas(fig)
-    p = librosa.display.specshow(librosa.amplitude_to_db(stft_hr), ax=ax1, y_axis='log', x_axis='time', sr=sr)
-    p = librosa.display.specshow(librosa.amplitude_to_db(stft_lr), ax=ax2, y_axis='log', x_axis='time', sr=sr)
-    p = librosa.display.specshow(librosa.amplitude_to_db(stft_recon), ax=ax3, y_axis='log', x_axis='time', sr=sr)
+    p = librosa.display.specshow(librosa.amplitude_to_db(stft_hr, ref=np.max), ax=ax1, y_axis='log', x_axis='time', sr=sr)
+    p = librosa.display.specshow(librosa.amplitude_to_db(stft_lr, ref=np.max), ax=ax2, y_axis='log', x_axis='time', sr=sr)
+    p = librosa.display.specshow(librosa.amplitude_to_db(stft_recon, ref=np.max), ax=ax3, y_axis='log', x_axis='time', sr=sr)
     return fig
 
 packet_size = CONFIG.DATA.EVAL.packet_size