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

Add diagonal to sparsity mask

Browse source
This commit is contained in:
Jean-Marc Valin 2018-10-15 02:45:27 -04:00
parent f13debcf65
commit fa1d2824fa
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
dnn/lpcnet.py
View file

@ -49,12 +49,14 @@ class Sparsify(Callback):
#print ("density = ", density)
for k in range(nb):
A = p[:, k*N:(k+1)*N]
A = A - np.diag(np.diag(A))
L=np.reshape(A, (N, N//16, 16))
S=np.sum(L*L, axis=-1)
SS=np.sort(np.reshape(S, (-1,)))
thresh = SS[round(N*N//16*(1-density))]
mask = (S>=thresh).astype('float32');
mask = np.repeat(mask, 16, axis=1)
mask = np.minimum(1, mask + np.diag(np.ones((N,))))
p[:, k*N:(k+1)*N] = p[:, k*N:(k+1)*N]*mask
#print(thresh, np.mean(mask))
w[1] = p