deep_filter 0.2.5

Noise supression using deep filtering
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

use std::env::args;
use std::process::exit;

use anyhow::Result;
use df::dataset::Hdf5Dataset;
use df::wav_utils::write_wav_arr2;
use rand::seq::SliceRandom;

fn main() -> Result<()> {
    let args = args().collect::<Vec<String>>();
    let p = match args.get(1) {
        Some(p) => p,
        None => {
            eprintln!("HDF5 dataset path expected");
            exit(1);
        }
    };
    let ds = Hdf5Dataset::new(p)?;
    let k = match args.get(2) {
        Some(k) => k.to_string(),
        None => ds.keys()?.choose(&mut rand::thread_rng()).unwrap().to_string(),
    };
    let data = ds.read(&k)?;
    println!("Sampled '{}' with audio shape {:?}", k, data.shape());
    let out_dir = args.get(3).cloned().unwrap_or_else(|| "out".to_owned());
    let name = format!("{}/{}", out_dir, k);
    write_wav_arr2(&name, data.view(), ds.sr.unwrap() as u32).unwrap();
    Ok(())
}