VigneshDark commited on
Commit
04bc49f
·
1 Parent(s): 5626d1e

removed snr

Browse files
Files changed (1) hide show
  1. helper.py +13 -13
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
- 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, speech_adjust + alpha)
90
- adjusted_noise = adjust_volume(trimmed_noise, noise_adjust + beta)
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