From f9fed09e96f43e7eab88378fc444c4987933badb Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 09 十二月 2022 23:57:51 +0800
Subject: [PATCH] Merge pull request #10 from alibaba-damo-academy/dev
---
egs/aishell/tranformer/utils/compute_cmvn.py | 11 +++++++++--
1 files changed, 9 insertions(+), 2 deletions(-)
diff --git a/egs/aishell/tranformer/utils/compute_cmvn.py b/egs/aishell/tranformer/utils/compute_cmvn.py
index 988d6dc..2b96e26 100755
--- a/egs/aishell/tranformer/utils/compute_cmvn.py
+++ b/egs/aishell/tranformer/utils/compute_cmvn.py
@@ -11,6 +11,13 @@
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
)
parser.add_argument(
+ "--dims",
+ "-d",
+ default=80,
+ type=int,
+ help="feature dims",
+ )
+ parser.add_argument(
"--ark-file",
"-a",
default=False,
@@ -44,8 +51,8 @@
ark_file = args.ark_file + "/feats." + str(args.ark_index) + ".ark"
cmvn_file = args.output_dir + "/cmvn." + str(args.ark_index) + ".json"
- mean_stats = 0.0
- var_stats = 0.0
+ mean_stats = np.zeros(args.dims)
+ var_stats = np.zeros(args.dims)
total_frames = 0
with ReadHelper('ark:{}'.format(ark_file)) as ark_reader:
--
Gitblit v1.9.1