eden-emu/opus
15
0
Fork 0
mirror of https://github.com/xiph/opus.git synced 2025-05-23 11:49:12 +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 17:14:29 -04:00
parent fd9002e98e
commit f884d7bf18
2 changed files with 12 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
dnn/lpcnet.py
View file

@ -16,15 +16,22 @@ nb_used_features = 37
def new_wavernn_model():
pcm = Input(shape=(None, 2))
pcm = Input(shape=(None, 1))
pitch = Input(shape=(None, 1))
feat = Input(shape=(None, nb_used_features))
conv1 = Conv1D(16, 7, padding='causal')
pconv1 = Conv1D(16, 5, padding='same')
pconv2 = Conv1D(16, 5, padding='same')
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()([pcm, rep(feat)])
rnn_in = Concatenate()([cpcm, cpitch, rep(feat)])
md = MDense(pcm_levels, activation='softmax')
ulaw_prob = md(rnn(rnn_in))
model = Model([pcm, feat], ulaw_prob)
model = Model([pcm, pitch, feat], ulaw_prob)
return model