mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-27 18:54:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

add kaldifeatwhisper fbank

Browse Source
This commit is contained in:
Yuekai Zhang 2024-01-23 21:22:47 +08:00
parent 08db3051ad
commit af29455c3d
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
egs/wenetspeech/ASR/local/compute_fbank_wenetspeech_splits.py
View File

@ -26,6 +26,8 @@ from lhotse import (
CutSet, CutSet,
WhisperFbank, WhisperFbank,
WhisperFbankConfig, WhisperFbankConfig,
KaldifeatWhisperFbank,
KaldifeatWhisperFbankConfig,
KaldifeatFbank, KaldifeatFbank,
KaldifeatFbankConfig, KaldifeatFbankConfig,
LilcomChunkyWriter, LilcomChunkyWriter,
@ -128,9 +130,10 @@ def compute_fbank_wenetspeech_splits(args):
if torch.cuda.is_available(): if torch.cuda.is_available():
device = torch.device("cuda", 0) device = torch.device("cuda", 0)
if args.whisper_fbank: if args.whisper_fbank:
extractor = WhisperFbank( # extractor = WhisperFbank(
WhisperFbankConfig(num_filters=args.num_mel_bins, device=device) # WhisperFbankConfig(num_filters=args.num_mel_bins, device=device)
) # )
extractor = KaldifeatWhisperFbank(KaldifeatWhisperFbankConfig(num_filters=args.num_mel_bins, device=device))
else: else:
extractor = KaldifeatFbank(KaldifeatFbankConfig(device=device)) extractor = KaldifeatFbank(KaldifeatFbankConfig(device=device))
logging.info(f"device: {device}") logging.info(f"device: {device}")