add FireRedVAD

2026-04-15 17:42:41 +08:00
parent 3e30360998
commit f901f4cbfe
22 changed files with 1244 additions and 37 deletions
@@ -0,0 +1,47 @@
+use aha::models::fire_red_vad::vad::FireRedVad;
+use anyhow::Result;
+
+#[test]
+fn aed() -> Result<()> {
+    // RUST_BACKTRACE=1 cargo test -F cuda --test test_fire_red_vad aed -r -- --nocapture
+    let audio_path = "file:///home/jhq/python_code/FireRedASR2S/assets/hello_zh.wav";
+    let device = aha::Device::Cpu;
+    let save_dir =
+        aha::utils::get_default_save_dir().ok_or(anyhow::anyhow!("Failed to get save dir"))?;
+    let model_path = format!("{}/xukaituo/FireRedVAD/AED/", save_dir);
+
+    let vad = FireRedVad::init(&model_path, Some(&device), None)?;
+    let res = vad.detect_file(audio_path)?;
+    println!("vad res: {:?}", res);
+    Ok(())
+}
+
+#[test]
+fn stream_vad() -> Result<()> {
+    // RUST_BACKTRACE=1 cargo test -F cuda --test test_fire_red_vad stream_vad -r -- --nocapture
+    let audio_path = "file:///home/jhq/python_code/FireRedASR2S/assets/hello_zh.wav";
+    let device = aha::Device::Cpu;
+    let save_dir =
+        aha::utils::get_default_save_dir().ok_or(anyhow::anyhow!("Failed to get save dir"))?;
+    let model_path = format!("{}/xukaituo/FireRedVAD/Stream-VAD/", save_dir);
+
+    let vad = FireRedVad::init(&model_path, Some(&device), None)?;
+    let res = vad.detect_file(audio_path)?;
+    println!("vad res: {:?}", res);
+    Ok(())
+}
+
+#[test]
+fn vad() -> Result<()> {
+    // RUST_BACKTRACE=1 cargo test -F cuda --test test_fire_red_vad vad -r -- --nocapture
+    let audio_path = "file:///home/jhq/python_code/FireRedASR2S/assets/hello_zh.wav";
+    let device = aha::Device::Cpu;
+    let save_dir =
+        aha::utils::get_default_save_dir().ok_or(anyhow::anyhow!("Failed to get save dir"))?;
+    let model_path = format!("{}/xukaituo/FireRedVAD/VAD/", save_dir);
+
+    let vad = FireRedVad::init(&model_path, Some(&device), None)?;
+    let res = vad.detect_file(audio_path)?;
+    println!("vad res: {:?}", res);
+    Ok(())
+}