eden-emu/opus
15
0
Fork 0
mirror of https://github.com/xiph/opus.git synced 2025-06-03 00:57:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add pitch feature computation

Browse source
This commit is contained in:
Jean-Marc Valin 2023-09-23 12:45:20 -04:00
parent 96d89e99d8
commit 0100cd95de
No known key found for this signature in database
GPG key ID: 531A52533318F00A
3 changed files with 57 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

22
dnn/dump_data.c
View file

@ -134,6 +134,7 @@ int main(int argc, char **argv) {
float noise_std=0;
int training = -1;
int burg = 0;
int pitch = 0;
srand(getpid());
st = lpcnet_encoder_create();
argv0=argv[0];
@ -145,6 +146,14 @@ int main(int argc, char **argv) {
burg = 1;
training = 0;
}
if (argc == 5 && strcmp(argv[1], "-ptrain")==0) {
pitch = 1;
training = 1;
}
if (argc == 4 && strcmp(argv[1], "-ptest")==0) {
pitch = 1;
training = 0;
}
if (argc == 5 && strcmp(argv[1], "-train")==0) training = 1;
if (argc == 4 && strcmp(argv[1], "-test")==0) training = 0;
if (training == -1) {
@ -239,7 +248,18 @@ int main(int argc, char **argv) {
compute_noise(noisebuf, noise_std);
}
process_single_frame(st, ffeat);
if (pitch) {
signed char pitch_features[PITCH_MAX_PERIOD-PITCH_MIN_PERIOD+PITCH_IF_FEATURES];
for (i=0;i<PITCH_MAX_PERIOD-PITCH_MIN_PERIOD;i++) {
pitch_features[i] = floor(.5 + 127.f*st->xcorr_features[i]);
}
for (i=0;i<PITCH_IF_FEATURES;i++) {
pitch_features[i+PITCH_MAX_PERIOD-PITCH_MIN_PERIOD] = floor(.5 + 127.f*st->if_features[i]);
}
fwrite(pitch_features, PITCH_MAX_PERIOD-PITCH_MIN_PERIOD+PITCH_IF_FEATURES, 1, ffeat);
} else {
process_single_frame(st, ffeat);
}
if (fpcm) write_audio(st, pcm, noisebuf, fpcm);
/*if (fpcm) fwrite(pcm, sizeof(opus_int16), FRAME_SIZE, fpcm);*/
for (i=0;i<TRAINING_OFFSET;i++) pcm[i] = float2short(x[i+FRAME_SIZE-TRAINING_OFFSET]);