joel/freqsplit
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: Corrected shape handling for separated sources, extracting each source as a separate 1D array

Browse Source
This commit is contained in:
Joel Mathew Thomas
2025-01-28 23:34:08 +05:30
parent 9e7c01c19e
commit fae60a8120
4 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test.py
+2 -3
View File
@@ -14,8 +14,7 @@ def main(input_audio_path, output_dir, model_name="mpariente/ConvTasNet_WHAM!_se
separated_sources = separate(audio, model_name)
print(f"Separated {len(separated_sources)} sources")
print("Separated sources are", separated_sources)
for i, source in enumerate(separated_sources):
output_path = os.path.join(output_dir, f"source_{i+1}.wav")
export_audio(source, output_path, sr)
@@ -26,7 +25,7 @@ def main(input_audio_path, output_dir, model_name="mpariente/ConvTasNet_WHAM!_se
if __name__ == "__main__":
input_audio_path = "/home/joel/Downloads/female_female_speech.wav"
input_audio_path = "/home/joel/Downloads/wham/female-female-mixture.wav"
output_dir = "/tmp/convtasnet"
main(input_audio_path, output_dir)