mirror of
https://github.com/xiph/opus.git
synced 2025-05-28 14:19:13 +00:00
Don't hardcode the number of bands
This commit is contained in:
parent
b5b1d5013e
commit
a9bf6cee8a
2 changed files with 5 additions and 3 deletions
|
@ -100,6 +100,8 @@ batch_size = args.batch_size
|
|||
quantize = args.quantize is not None
|
||||
retrain = args.retrain is not None
|
||||
|
||||
lpc_order = 16
|
||||
|
||||
if quantize:
|
||||
lr = 0.00003
|
||||
decay = 0
|
||||
|
@ -133,7 +135,7 @@ with strategy.scope():
|
|||
feature_file = args.features
|
||||
pcm_file = args.data # 16 bit unsigned short PCM samples
|
||||
frame_size = model.frame_size
|
||||
nb_features = 36
|
||||
nb_features = model.nb_used_features + lpc_order
|
||||
nb_used_features = model.nb_used_features
|
||||
feature_chunk_size = 15
|
||||
pcm_chunk_size = frame_size*feature_chunk_size
|
||||
|
@ -160,7 +162,7 @@ features = np.lib.stride_tricks.as_strided(features, shape=(nb_frames, feature_c
|
|||
#features = features[:, :, :nb_used_features]
|
||||
|
||||
|
||||
periods = (.1 + 50*features[:,:,18:19]+100).astype('int16')
|
||||
periods = (.1 + 50*features[:,:,nb_used_features-2:nb_used_features-1]+100).astype('int16')
|
||||
#periods = np.minimum(periods, 255)
|
||||
|
||||
# dump models to disk as we go
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue