implement a lms filter for echo cancellation from audio files in python

main.py
import numpy as np
import soundfile as sf
42 chars
3 lines

main.py
audio, sample_rate = sf.read('input_audio.wav')
48 chars
2 lines

main.py
order = 64  # Order of the LMS filter
mu = 0.01   # Step size (learning rate)
78 chars
3 lines

main.py
weights = np.zeros(order)
delay_line = np.zeros(order)
55 chars
3 lines

main.py
filtered_audio = np.zeros(len(audio))
for i in range(len(audio)):
    # Filter the audio sample
    y = np.dot(weights, delay_line)
    
    # Calculate the error (difference between the filtered and original audio)
    error = audio[i] - y
    
    # Update the filter coefficients using the LMS algorithm
    weights += 2 * mu * error * delay_line
    
    # Update the delay line
    delay_line = np.roll(delay_line, 1)
    delay_line[0] = audio[i]
    
    # Store the filtered audio sample
    filtered_audio[i] = y
521 chars
18 lines

main.py
sf.write('filtered_audio.wav', filtered_audio, sample_rate)
60 chars
2 lines