fix incorrect RMS calculation - issue #44

This commit is contained in:
Sergiy Matusevych 2021-11-11 07:50:40 -08:00
Родитель 5c0d72e4db
Коммит 76ea71d6bb
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -202,7 +202,7 @@ def active_rms(clean, noise, fs=16000, energy_thresh=-50):
sample_end = min(sample_start + window_samples, len(noise))
noise_win = noise[sample_start:sample_end]
clean_win = clean[sample_start:sample_end]
noise_seg_rms = 20*np.log10((noise_win**2).mean()+EPS)
noise_seg_rms = (noise_win**2).mean()**0.5
# Considering frames with energy
if noise_seg_rms > energy_thresh:
noise_active_segs = np.append(noise_active_segs, noise_win)