Commit graph

96 commits

Author SHA1 Message Date
Jan Buethe
367a487e7c
hangover fix in osce/utils/pitch.py 2024-02-15 15:39:30 +01:00
Jan Buethe
735117b6d7
disabled sparse option in osce export script 2024-02-15 15:25:06 +01:00
Jan Buethe
7df2c67be1
fixes in osce python code 2024-01-23 17:10:34 +01:00
Jan Buethe
ec04a94eb2
bugfix in SilkFeatureNetPL 2024-01-22 15:12:52 +01:00
Jean-Marc Valin
1ddfcfd48c
Using PyTorch model (same architecture for now) 2024-01-21 02:11:50 -05:00
Jan Buethe
299e38cab7
Updated LACE and NoLACE models to version 2 2024-01-20 14:44:22 +01:00
Jean-Marc Valin
4f311a1ad4
PLC export script
mostly untested
2024-01-17 02:26:48 -05:00
Jean-Marc Valin
26ddfd7135
PyTorch code for training the PLC model
Should match the TF2 code, but mostly untested
2024-01-15 18:11:47 -05:00
Jean-Marc Valin
bd2e9a34fb
Add simulated loss to opus_demo 2023-12-22 03:42:16 -05:00
Jean-Marc Valin
bd710e97f3
C code for packet loss simulator 2023-12-21 21:30:53 -05:00
Jean-Marc Valin
b923fd1e28
lossgen: better training, README.md 2023-12-21 18:01:57 -05:00
Jean-Marc Valin
c40add59af
lossgen: can now dump weights 2023-12-21 16:57:35 -05:00
Jean-Marc Valin
627aa7f5b3
Packet loss generation model 2023-12-21 15:34:33 -05:00
Jan Buethe
7d328f5bfa
Merge LACE/NoLACE under OSCE framework 2023-12-20 03:42:44 -05:00
Jean-Marc Valin
7f7b2a1c66
Smaller version of fargan
800k parameters, 600 MFLOPS, with a receptive field of 3 feature vectors
2023-11-16 02:06:14 -05:00
Jean-Marc Valin
b0620c0bf9
Using sparse GRUs in DRED decoder
Saves ~270 kB of weights in the decoder
2023-11-15 04:08:50 -05:00
Jean-Marc Valin
77594bf158
Dumping RDOVAE stats from XML 2023-11-08 17:32:43 -05:00
Jean-Marc Valin
222662dac8
DRED: quantize scale and dead zone to 8 bits 2023-11-07 18:10:50 -05:00
Jan Buethe
4e104555e9
added weight export script for LACE/NoLACE 2023-11-07 15:12:12 +01:00
Jan Buethe
8af5c6b4a1
added transposed 1d convolutions to wexchange 2023-11-07 11:54:22 +01:00
Jean-Marc Valin
0ab0640d4a
Split stats in two and remove useless dimensions 2023-11-07 00:07:14 -05:00
Jan Buethe
2386a60ec6
updated moc to match results in ietf118 presentation 2023-11-06 17:50:48 +01:00
Jean-Marc Valin
544b3e576c
DRED: quantize r and p0 parameters with 8 bits
Only code non-degenerate symbols, which makes the encoder faster
2023-11-06 03:16:43 -05:00
Jan Buethe
da60266f6e
updated moc method 2023-11-02 16:52:50 +01:00
Jean-Marc Valin
ddd5669e79
Pitch and fargan model updates
Removing one of the 2d conv layers for pitch estimation reduces
complexity without noticeable degradation. FARGAN model has more
adversarial training.
Also, no need for the double precision in the low-pass filter.
2023-10-28 23:33:47 -04:00
Jean-Marc Valin
4598fe5409
Quantizing pitchdnn and rdovae weights 2023-10-20 12:54:13 -04:00
Jan Buethe
290be25b98
added 16kHz version of opus_compare in python 2023-10-20 14:24:27 +02:00
Jan Buethe
1accd2472e
finalized quantization option in export_rdovae_weights.py 2023-10-20 14:14:31 +02:00
Jean-Marc Valin
d720955d61
Marking RDOVAE layers to quantize 2023-10-19 16:06:52 -04:00
Jan Buethe
60ac1c6c99
prepared quantization implementation for DRED 2023-10-19 21:54:39 +02:00
Jan Buethe
2192e85b91
restructured osce readme 2023-10-19 21:45:45 +02:00
Jan Buethe
055c683018
added LACE/NoLACE checkpoint URL 2023-10-19 21:34:13 +02:00
Jean-Marc Valin
35cb8d7f66
C implementation of FARGAN 2023-10-14 16:42:33 -04:00
Jean-Marc Valin
9e76a7bfb8
update fargan to match version 45 2023-10-10 00:51:57 -04:00
Jan Buethe
0563d71b25
updated osce readme 2023-10-07 18:52:38 +02:00
Jean-Marc Valin
f0ec990dba
Switching to neural pitch estimator
Remove old pitch estimator and retrain all models
2023-10-06 03:14:56 -04:00
Jean-Marc Valin
27663d3641
Using a DenseNet for DRED 2023-10-02 01:43:44 -04:00
Jean-Marc Valin
8e8edf71bd
Remove unneeded (I think) tanh at the end 2023-10-01 21:34:58 -04:00
Jean-Marc Valin
33adba02c7
First version of pitch DNN C code
Totally untested -- most likely doesn't work
2023-10-01 03:59:17 -04:00
Jean-Marc Valin
f3b86f9414
Fix model saving 2023-09-30 02:48:26 -04:00
Jan Buethe
0459a572f5
updated PitchDNN export script 2023-09-29 15:34:59 +02:00
Jan Buethe
ce28695844
refactoring and cleanup 2023-09-29 15:31:45 +02:00
Jan Buethe
4901445490
fixed type in error message 2023-09-29 14:34:11 +02:00
Jan Buethe
c5c214df1b
added rudimentary support for dumping nn.Conv2d layers 2023-09-29 14:25:26 +02:00
Jean-Marc Valin
9a7bb764d4
No features skip needed to align pitch features 2023-09-27 13:01:21 -04:00
Jean-Marc Valin
a6b4fe375a
Script to compute the groundtruth data using CREPE 2023-09-27 13:00:12 -04:00
Jan Buethe
ae0a140c40
updated default values in adv_train_fargan.py 2023-09-26 21:42:01 +02:00
Jan Buethe
41a4c9515d
changed checkpoint format 2023-09-26 12:12:47 -04:00
Jean-Marc Valin
733a095ba2
Adapting to new data format/model 2023-09-26 12:12:47 -04:00
Krishna Subramani
f38b4a317f
Python code for neural pitch 2023-09-26 12:12:47 -04:00