From adcee8828ef5d78b575043954deb662a35e318f7 Mon Sep 17 00:00:00 2001
From: huangmingming <huangmingming@deepscience.cn>
Date: 星期一, 30 一月 2023 16:02:54 +0800
Subject: [PATCH] update the minimum size of audio
---
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