From 5f3f194ffd459fd5ebe6ea46da5e532820a66060 Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期五, 21 七月 2023 11:24:31 +0800
Subject: [PATCH] update
---
egs/callhome/eend_ola/local/dump_feature.py | 18 +++++++++---------
1 files changed, 9 insertions(+), 9 deletions(-)
diff --git a/egs/callhome/eend_ola/local/dump_feature.py b/egs/callhome/eend_ola/local/dump_feature.py
index 332edd2..8549c31 100644
--- a/egs/callhome/eend_ola/local/dump_feature.py
+++ b/egs/callhome/eend_ola/local/dump_feature.py
@@ -34,7 +34,7 @@
utt2spk_file = os.path.join(self.data_dir, 'utt2spk.{}'.format(idx))
self.utt2spk = load_utt2spk(utt2spk_file)
- wav_file = os.path.join(self.data_dir, 'wav.scp.{}'.format(idx))
+ wav_file = os.path.join(self.data_dir, 'wav.{}.scp'.format(idx))
self.wavs = load_wav_scp(wav_file)
reco2dur_file = os.path.join(self.data_dir, 'reco2dur.{}'.format(idx))
@@ -131,14 +131,14 @@
if __name__ == '__main__':
parser = argparse.ArgumentParser()
- parser.add_argument("data_dir")
- parser.add_argument("output_dir")
- parser.add_argument("index")
- parser.add_argument("num_frames", default=500)
- parser.add_argument("context_size", default=7)
- parser.add_argument("frame_size", default=200)
- parser.add_argument("frame_shift", default=80)
- parser.add_argument("subsampling", default=10)
+ parser.add_argument("--data_dir", type=str)
+ parser.add_argument("--output_dir", type=str)
+ parser.add_argument("--index", type=str)
+ parser.add_argument("--num_frames", type=int, default=500)
+ parser.add_argument("--context_size", type=int, default=7)
+ parser.add_argument("--frame_size", type=int, default=200)
+ parser.add_argument("--frame_shift", type=int, default=80)
+ parser.add_argument("--subsampling", type=int, default=10)
args = parser.parse_args()
convert(args)
--
Gitblit v1.9.1