From 7ff1fbe2becbc7b5a3597c2bd71981d32b9af566 Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 13 Jun 2022 20:37:35 -0600 Subject: [PATCH] channel clipper --- codes/trainer/injectors/audio_injectors.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/codes/trainer/injectors/audio_injectors.py b/codes/trainer/injectors/audio_injectors.py index b6d8529c..dae5c84a 100644 --- a/codes/trainer/injectors/audio_injectors.py +++ b/codes/trainer/injectors/audio_injectors.py @@ -368,5 +368,16 @@ class NormalizeMelInjector(Injector): return {self.output: normalize_mel(mel)} +class ChannelClipInjector(Injector): + def __init__(self, opt, env): + super().__init__(opt, env) + self.lo = opt['channel_low'] + self.hi = opt['channel_high'] + + def forward(self, state): + inp = state[self.input] + return {self.output: inp[:,self.lo:self.hi]} + + if __name__ == '__main__': print('hi') \ No newline at end of file