From 9e366e76ba14b20a9d153e96337ef3b307f4065d Mon Sep 17 00:00:00 2001 From: Mart Lubbers Date: Wed, 7 Jun 2017 16:37:35 +0200 Subject: [PATCH] predict smoothing --- predict.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/predict.py b/predict.py index de1abc2..3e0e2f7 100644 --- a/predict.py +++ b/predict.py @@ -38,13 +38,13 @@ tier = tgob.add_tier('lyrics') window_len = int(1.0/winstep) x = model.predict(data, batch_size=32, verbose=0) -#s = np.r_[x[window_len-1:0:-1],x,x[-2:-window_len-1:-1]] -#w = np.hanning(window_len) +s = np.r_[x[window_len-1:0:-1],x,x[-2:-window_len-1:-1]] +w = np.hanning(window_len) # -#smoothed = np.convolve(w/w.sum(), s[:,0], mode='valid') -#wavdata = np.uint8(list(map(int, -# smoothed*255))[int(window_len/2):-1*(int(window_len/2))]) -wavdata = np.uint8(x*255) +smoothed = np.convolve(w/w.sum(), s[:,0], mode='valid') +wavdata = np.uint8(list(map(int, + smoothed*255))[int(window_len/2):-1*(int(window_len/2))]) +#wavdata = np.uint8(x*255) print('sr: ', int(1.0/winstep)) print("len(wavdata): ", len(wavdata)) -- 2.20.1