eden-emu/opus
15
0
Fork 0
mirror of https://github.com/xiph/opus.git synced 2025-05-19 09:58:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add convolution

Browse source
This commit is contained in:
Jean-Marc Valin 2018-06-26 22:52:24 -04:00
parent f884d7bf18
commit 06511ba5a4
2 changed files with 13 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
dnn/lpcnet.py
View file

@ -23,13 +23,22 @@ def new_wavernn_model():
conv1 = Conv1D(16, 7, padding='causal')
pconv1 = Conv1D(16, 5, padding='same')
pconv2 = Conv1D(16, 5, padding='same')
fconv1 = Conv1D(128, 3, padding='same')
fconv2 = Conv1D(32, 3, padding='same')
if True:
cpcm = conv1(pcm)
cpitch = pconv2(pconv1(pitch))
else:
cpcm = pcm
cpitch = pitch
cfeat = fconv2(fconv1(feat))
cpcm = conv1(pcm)
cpitch = pconv2(pconv1(pitch))
rep = Lambda(lambda x: K.repeat_elements(x, 160, 1))
rnn = CuDNNGRU(rnn_units, return_sequences=True)
rnn_in = Concatenate()([cpcm, cpitch, rep(feat)])
rnn_in = Concatenate()([cpcm, cpitch, rep(cfeat)])
md = MDense(pcm_levels, activation='softmax')
ulaw_prob = md(rnn(rnn_in))