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
| import argparse
| import json
| import numpy as np
|
| def get_parser():
| parser = argparse.ArgumentParser(
| description="combine cmvn file",
| formatter_class=argparse.ArgumentDefaultsHelpFormatter,
| )
| parser.add_argument(
| "--dims",
| "-d",
| default=80,
| type=int,
| help="feature dims",
| )
| parser.add_argument(
| "--cmvn-dir",
| "-c",
| default=False,
| required=True,
| type=str,
| help="cmvn dir",
| )
|
| parser.add_argument(
| "--nj",
| "-n",
| default=1,
| required=True,
| type=int,
| help="num of cmvn file",
| )
| parser.add_argument(
| "--output-dir",
| "-o",
| default=False,
| required=True,
| type=str,
| help="output dir",
| )
| return parser
|
|
| def main():
| parser = get_parser()
| args = parser.parse_args()
|
| total_means = np.zeros(args.dims)
| total_vars = np.zeros(args.dims)
| total_frames = 0
|
| cmvn_file = args.output_dir + "/cmvn.json"
|
| for i in range(1, args.nj+1):
| with open(args.cmvn_dir + "/cmvn." + str(i) + ".json", "r") as fin:
| cmvn_stats = json.load(fin)
|
| total_means += np.array(cmvn_stats["mean_stats"])
| total_vars += np.array(cmvn_stats["var_stats"])
| total_frames += cmvn_stats["total_frames"]
|
| cmvn_info = {
| 'mean_stats': list(total_means.tolist()),
| 'var_stats': list(total_vars.tolist()),
| 'total_frames': total_frames
| }
| with open(cmvn_file, 'w') as fout:
| fout.write(json.dumps(cmvn_info))
|
|
| if __name__ == '__main__':
| main()
|
|
