eden-emu/opus
15
0
Fork 0
mirror of https://github.com/xiph/opus.git synced 2025-05-18 17:38:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

...

Browse source
This commit is contained in:
Jean-Marc Valin 2018-10-16 04:05:52 -04:00
parent fa1d2824fa
commit fb1d4fdec2
2 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
dnn/lpcnet.py
View file

@ -11,7 +11,7 @@ import numpy as np
import h5py import h5py
import sys import sys
rnn_units1=256 rnn_units1=512
rnn_units2=32 rnn_units2=32
pcm_bits = 8 pcm_bits = 8
embed_size = 128 embed_size = 128

7
dnn/train_wavenet_audio.py
View file

@ -111,6 +111,7 @@ in_exc = in_exc.astype('uint8')
features = np.reshape(features, (nb_frames, feature_chunk_size, nb_features)) features = np.reshape(features, (nb_frames, feature_chunk_size, nb_features))
features = features[:, :, :nb_used_features] features = features[:, :, :nb_used_features]
features[:,:,18:36] = 0
pred = np.reshape(pred, (nb_frames, pcm_chunk_size, 1)) pred = np.reshape(pred, (nb_frames, pcm_chunk_size, 1))
pred = pred.astype('uint8') pred = pred.astype('uint8')
@ -119,8 +120,8 @@ periods = (50*features[:,:,36:37]+100).astype('int16')
in_data = np.concatenate([in_data, pred], axis=-1) in_data = np.concatenate([in_data, pred], axis=-1)
# dump models to disk as we go # dump models to disk as we go
checkpoint = ModelCheckpoint('wavenet5p0_{epoch:02d}.h5') checkpoint = ModelCheckpoint('lpcnet5_512_10_G32np_{epoch:02d}.h5')
#model.load_weights('wavenet4f2_30.h5') #model.load_weights('wavenet4f2_30.h5')
model.compile(optimizer=Adam(0.001, amsgrad=True, decay=5e-5), loss='sparse_categorical_crossentropy', metrics=['sparse_categorical_accuracy']) model.compile(optimizer=Adam(0.0005, amsgrad=True, decay=5e-5), loss='sparse_categorical_crossentropy', metrics=['sparse_categorical_accuracy'])
model.fit([in_data, in_exc, features, periods], out_data, batch_size=batch_size, epochs=60, validation_split=0.2, callbacks=[checkpoint, lpcnet.Sparsify(1000, 20000, 200, 0.25)]) model.fit([in_data, in_exc, features, periods], out_data, batch_size=batch_size, epochs=60, validation_split=0.2, callbacks=[checkpoint, lpcnet.Sparsify(1000, 20000, 200, 0.1)])