Skip to content

fix hptsm for multichannel audio #8

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
seyong92 merged 1 commit into from
Oct 6, 2020
Merged

fix hptsm for multichannel audio #8

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 12 additions & 3 deletions pytsmod/hptsm.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
from librosa.effects import hpss
import numpy as np
from .pvtsm import phase_vocoder
from .olatsm import ola
from .utils import _validate_audio


def hptsm(x, s, hp_kernel_size=31, hp_power=2.0, hp_mask=False, hp_margin=1.0,
Expand Down Expand Up @@ -32,9 +34,16 @@ def hptsm(x, s, hp_kernel_size=31, hp_power=2.0, hp_mask=False, hp_margin=1.0,
y : numpy.ndarray [shape=(channel, num_samples) or (num_samples)]
the modified output audio sequence.
"""

x_harm, x_perc = hpss(x, kernel_size=hp_kernel_size, power=hp_power,
mask=hp_mask, margin=hp_margin)
x = _validate_audio(x)
x_harm = np.zeros(x.shape)
x_perc = np.zeros(x.shape)

for c, x_chan in enumerate(x):
x_harm_chan, x_perc_chan = hpss(x_chan, kernel_size=hp_kernel_size,
power=hp_power, mask=hp_mask,
margin=hp_margin)
x_harm[c, :] = x_harm_chan
x_perc[c, :] = x_perc_chan

y_harm = phase_vocoder(x_harm, s, win_type=pv_win_type,
win_size=pv_win_size, syn_hop_size=pv_syn_hop_size,
Expand Down