| | from nara_wpe.wpe import wpe |
| |
|
| |
|
| | class WPE(object): |
| | def __init__( |
| | self, taps=10, delay=3, iterations=3, psd_context=0, statistics_mode="full" |
| | ): |
| | self.taps = taps |
| | self.delay = delay |
| | self.iterations = iterations |
| | self.psd_context = psd_context |
| | self.statistics_mode = statistics_mode |
| |
|
| | def __repr__(self): |
| | return ( |
| | "{name}(taps={taps}, delay={delay}" |
| | "iterations={iterations}, psd_context={psd_context}, " |
| | "statistics_mode={statistics_mode})".format( |
| | name=self.__class__.__name__, |
| | taps=self.taps, |
| | delay=self.delay, |
| | iterations=self.iterations, |
| | psd_context=self.psd_context, |
| | statistics_mode=self.statistics_mode, |
| | ) |
| | ) |
| |
|
| | def __call__(self, xs): |
| | """Return enhanced |
| | |
| | :param np.ndarray xs: (Time, Channel, Frequency) |
| | :return: enhanced_xs |
| | :rtype: np.ndarray |
| | |
| | """ |
| | |
| | xs = wpe( |
| | xs.transpose((2, 1, 0)), |
| | taps=self.taps, |
| | delay=self.delay, |
| | iterations=self.iterations, |
| | psd_context=self.psd_context, |
| | statistics_mode=self.statistics_mode, |
| | ) |
| | return xs.transpose(2, 1, 0) |
| |
|
