From 961c6de3a54e94c94d41dfefea7bffcdc362c159 Mon Sep 17 00:00:00 2001 From: jhqxxx <18280426169@163.com> Date: Thu, 16 Apr 2026 20:05:06 +0800 Subject: [PATCH] fix fire_red_vad_fbank frame num bug --- src/models/fire_red_vad/processor.rs | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/models/fire_red_vad/processor.rs b/src/models/fire_red_vad/processor.rs index 48b436e..145b39d 100644 --- a/src/models/fire_red_vad/processor.rs +++ b/src/models/fire_red_vad/processor.rs @@ -4,7 +4,7 @@ use anyhow::{Result, anyhow}; use candle_core::{D, Device, IndexOp, Tensor}; use kaldi_native_fbank::{ FbankComputer, FbankOptions, - window::{Window, extract_window}, + window::{Window, extract_window, num_frames}, }; use crate::{ @@ -94,7 +94,8 @@ impl KaldifeatFbank { }; let mut feats = vec![]; let mut window_buf = vec![0.0; padded]; - for frame in 0..230 { + let frame_len = num_frames(wave.len(), &self.opts.frame_opts, true); + for frame in 0..frame_len { let raw_log_energy = extract_window( 0, &wave,