--- license: apache-2.0 pipeline_tag: audio-to-audio tags: - speech_enhancement - noise_suppression - real_time - streaming - causal - onnx - tflite - fullband --- # DPDFNet DPDFNet is a family of **causal, single‑channel** speech enhancement models for **real‑time noise suppression**.\ It builds on **DeepFilterNet2** by adding **Dual‑Path RNN (DPRNN)** blocks in the encoder for stronger long‑range modeling while staying streaming‑friendly. **Links** - Project page (audio samples + architecture): https://ceva-ip.github.io/DPDFNet/ - Paper (arXiv): https://arxiv.org/abs/2512.16420 - Code (GitHub): https://github.com/ceva-ip/DPDFNet - Demo Space: https://huggingface.co/spaces/Ceva-IP/DPDFNetDemo - Evaluation set: https://huggingface.co/datasets/Ceva-IP/DPDFNet_EvalSet --- ## What’s in this repo - **TFLite**: `*.tflite` (root) - **ONNX**: `onnx/*.onnx` - **PyTorch checkpoints**: `checkpoints/*.pth` --- ## Model variants ### 16 kHz models | Model | DPRNN blocks | Params (M) | MACs (G) | |---|:---:|:---:|:---:| | `baseline` | 0 | 2.31 | 0.36 | | `dpdfnet2` | 2 | 2.49 | 1.35 | | `dpdfnet4` | 4 | 2.84 | 2.36 | | `dpdfnet8` | 8 | 3.54 | 4.37 | ### 48 kHz fullband model | Model | DPRNN blocks | Params (M) | MACs (G) | |---|:---:|:---:|:---:| | `dpdfnet2_48khz_hr` | 2 | 2.58 | 2.42 | --- ## Recommended inference (CPU-only, ONNX) ```bash pip install dpdfnet ``` ### CLI ```bash # Enhance one file dpdfnet enhance noisy.wav enhanced.wav --model dpdfnet4 # Enhance a directory dpdfnet enhance-dir ./noisy_wavs ./enhanced_wavs --model dpdfnet2 # Download models dpdfnet download dpdfnet download dpdfnet8 dpdfnet download dpdfnet4 --force ``` ### Python API ```python import soundfile as sf import dpdfnet # In-memory enhancement: audio, sr = sf.read("noisy.wav") enhanced = dpdfnet.enhance(audio, sample_rate=sr, model="dpdfnet4") sf.write("enhanced.wav", enhanced, sr) # Enhance one file: out_path = dpdfnet.enhance_file("noisy.wav", model="dpdfnet2") print(out_path) # Model listing: for row in dpdfnet.available_models(): print(row["name"], row["ready"], row["cached"]) # Download models: dpdfnet.download() # All models dpdfnet.download("dpdfnet4") # Specific model ``` --- ## Citation ```bibtex @article{rika2025dpdfnet, title = {DPDFNet: Boosting DeepFilterNet2 via Dual-Path RNN}, author = {Rika, Daniel and Sapir, Nino and Gus, Ido}, year = {2025} } ``` --- ## License Apache-2.0