From a0ef664ee2bbf7251af06d485f7c1c45efa4f343 Mon Sep 17 00:00:00 2001
From: Chin-Yun Yu <cy006@wynne.eecs.qmul.ac.uk>
Date: Wed, 3 May 2023 16:24:45 +0100
Subject: [PATCH] fix: mono speech + average
---
my_submission/ensemble.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/my_submission/ensemble.py b/my_submission/ensemble.py
index 907da47..60bd8a7 100644
--- a/my_submission/ensemble.py
+++ b/my_submission/ensemble.py
@@ -161,7 +161,7 @@ class EnsembleNet(IdentitySeparationModel):
separated_music_arrays["effect"] += separated_fx
separated_music_arrays["effect"] /= 2
- separated_music_arrays["dialog"] = separated_speech
+ separated_music_arrays["dialog"] += separated_speech.mean(axis=1, keepdims=True)
separated_music_arrays["dialog"] /= 2
return separated_music_arrays, output_sample_rates
--
GitLab