48 lines
1.8 KiB
Rust
48 lines
1.8 KiB
Rust
|
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(())
|
||
|
|
}
|