mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-27 10:44:19 +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

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

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