Commit
·
04bc49f
1
Parent(s):
5626d1e
removed snr
Browse files
helper.py
CHANGED
@@ -71,23 +71,23 @@ def process_audio(speech_data, noise_data, speech_sr, noise_sr, alpha, beta):
|
|
71 |
|
72 |
trimmed_noise = trimmed_noise.set_frame_rate(8000)
|
73 |
|
74 |
-
# Calculate volumes and adjustments
|
75 |
-
speech_vol = get_audio_volume_db(speech_audio)
|
76 |
-
noise_vol = get_audio_volume_db(trimmed_noise)
|
77 |
|
78 |
-
current_snr = speech_vol - noise_vol
|
79 |
-
adjustment_needed = 10 - current_snr # target_snr hardcoded to 10
|
80 |
|
81 |
-
if adjustment_needed > 0: # Speech too quiet
|
82 |
-
|
83 |
-
|
84 |
-
else: # Speech too loud
|
85 |
-
|
86 |
-
|
87 |
|
88 |
# Apply adjustments
|
89 |
-
adjusted_speech = adjust_volume(speech_audio,
|
90 |
-
adjusted_noise = adjust_volume(trimmed_noise,
|
91 |
|
92 |
final_audio = overlay_audio(adjusted_speech, adjusted_noise)
|
93 |
|
|
|
71 |
|
72 |
trimmed_noise = trimmed_noise.set_frame_rate(8000)
|
73 |
|
74 |
+
# # Calculate volumes and adjustments
|
75 |
+
# speech_vol = get_audio_volume_db(speech_audio)
|
76 |
+
# noise_vol = get_audio_volume_db(trimmed_noise)
|
77 |
|
78 |
+
# current_snr = speech_vol - noise_vol
|
79 |
+
# adjustment_needed = 10 - current_snr # target_snr hardcoded to 10
|
80 |
|
81 |
+
# if adjustment_needed > 0: # Speech too quiet
|
82 |
+
# speech_adjust = min(adjustment_needed, 2)
|
83 |
+
# noise_adjust = -min(adjustment_needed / 2, 5)
|
84 |
+
# else: # Speech too loud
|
85 |
+
# speech_adjust = max(adjustment_needed, -5)
|
86 |
+
# noise_adjust = -5 / 2
|
87 |
|
88 |
# Apply adjustments
|
89 |
+
adjusted_speech = adjust_volume(speech_audio, alpha)
|
90 |
+
adjusted_noise = adjust_volume(trimmed_noise, beta)
|
91 |
|
92 |
final_audio = overlay_audio(adjusted_speech, adjusted_noise)
|
93 |
|