1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
/*
extract.c
david Rowe Jan 2019
Extracts sub sets of vectors from .f32 files, used for LPCNet VQ experiments.
*/
#include <assert.h>
#include <stdlib.h>
#include <stdio.h>
#include <getopt.h>
#define NB_FEATURES 55 /* number of cols per row */
int main(int argc, char *argv[]) {
FILE *fin, *fout;
int st = 0;
int en = 17;
int stride = NB_FEATURES;
float gain = 1.0;
int frame_delay = 1;
float pred = 0.0;
int removemean = 0;
float lower = -1E32;
static struct option long_options[] = {
{"startcol", required_argument, 0, 's'},
{"endcol", required_argument, 0, 'e'},
{"stride", required_argument, 0, 't'},
{"gain", required_argument, 0, 'g'},
{"pred", required_argument, 0, 'p'},
{"delay", required_argument, 0, 'd'},
{"removemean", no_argument, 0, 'm'},
{"lower", required_argument, 0, 'l'},
{0, 0, 0, 0}
};
int opt_index = 0;
int c;
while ((c = getopt_long (argc, argv, "s:e:t:g:p:d:ml:", long_options, &opt_index)) != -1) {
switch (c) {
case 's':
st = atoi(optarg);
break;
case 'e':
en = atoi(optarg);
break;
case 't':
stride = atoi(optarg);
break;
case 'g':
gain = atof(optarg);
break;
case 'p':
pred = atof(optarg);
break;
case 'd':
frame_delay = atoi(optarg);
break;
case 'm':
removemean = 1;
break;
case 'l':
lower = atof(optarg);
break;
default:
helpmsg:
fprintf(stderr, "usage: %s -s startCol -e endCol [-t strideCol -g gain -p predCoeff -d framesDelay --removemean --lower] input.f32 output.f32\n", argv[0]);
exit(1);
}
}
if ( (argc - optind) < 2) {
fprintf(stderr, "Too few arguments\n");
goto helpmsg;
}
fin = fopen(argv[optind],"rb"); assert(fin != NULL);
fout = fopen(argv[optind+1],"wb"); assert(fout != NULL);
printf("extracting from %d to %d inclusive (stride %d) ... gain = %f pred = %f frame_delay = %d\n",
st, en, stride, gain, pred, frame_delay);
float features[stride], features_prev[frame_delay][stride], delta[stride];
int i,f,wr=0;
for (f=0; f<frame_delay; f++)
for(i=0; i<stride; i++)
features_prev[f][i] = 0.0;
while((fread(features, sizeof(float), stride, fin) == stride)) {
float mean = 0.0;
for(i=st; i<=en; i++)
mean += features[i];
mean /= (en-st+1);
if (removemean) {
for(i=st; i<=en; i++)
features[i] -= mean;
}
for(i=st; i<=en; i++) {
delta[i] = gain*(features[i] - pred*features_prev[frame_delay-1][i]);
}
if (mean > lower) {
fwrite(&delta[st], sizeof(float), en-st+1, fout);
wr++;
}
for (f=frame_delay-1; f>0; f--)
for(i=0; i<stride; i++)
features_prev[f][i] = features_prev[f-1][i];
for(i=0; i<stride; i++)
features_prev[0][i] = features[i];
}
fclose(fin); fclose(fout);
fprintf(stderr, "%d extracted\n", wr);
return 0;
}
|
