Skip to main content

CPU-only ONNX inference package for DPDFNet speech enhancement.

Project description

dpdfnet

CPU-only ONNX inference package for DPDFNet speech enhancement.

Installation

pip install dpdfnet

Requirements

  • Python >=3.11
  • OS support for soundfile / libsndfile

Runtime dependencies are installed automatically:

  • numpy
  • librosa
  • soundfile
  • onnxruntime
  • filelock
  • tqdm

CLI

Show help:

dpdfnet --help

Commands:

  1. dpdfnet models
  • List supported models and local availability.
  1. dpdfnet enhance <input.wav> <output.wav> [--model <name>] [-v|--verbose]
  • Enhance one WAV file.
  1. dpdfnet enhance-dir <input_dir> <output_dir> [--model <name>] [-v|--verbose]
  • Enhance all .wav files in a directory (non-recursive).
  1. dpdfnet download [model] [--force|--refresh] [-q|--quiet | -v|--verbose]
  • Download all models when model is omitted, or one model when provided.

CLI examples:

# 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

Top-level exports:

  • dpdfnet.enhance
  • dpdfnet.enhance_file
  • dpdfnet.available_models
  • dpdfnet.download

In-memory enhancement:

import soundfile as sf
import dpdfnet

audio, sr = sf.read("noisy.wav")
enhanced = dpdfnet.enhance(audio, sample_rate=sr, model="dpdfnet4")
sf.write("enhanced.wav", enhanced, sr)

Enhance one file:

import dpdfnet

out_path = dpdfnet.enhance_file("noisy.wav", model="dpdfnet2")
print(out_path)

Model listing:

import dpdfnet

for row in dpdfnet.available_models():
    print(row["name"], row["ready"], row["cached"])

Download models via API:

import dpdfnet

dpdfnet.download()
dpdfnet.download("dpdfnet4")

Links

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

dpdfnet-0.3.0.tar.gz (18.5 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

dpdfnet-0.3.0-py3-none-any.whl (17.8 kB view details)

Uploaded Python 3

File details

Details for the file dpdfnet-0.3.0.tar.gz.

File metadata

  • Download URL: dpdfnet-0.3.0.tar.gz
  • Upload date:
  • Size: 18.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.14

File hashes

Hashes for dpdfnet-0.3.0.tar.gz
Algorithm Hash digest
SHA256 20208f0f0e20c9d5a0911b99a750232287ebe7cb35d6bc0951d530728febc52b
MD5 9eb81ba65b98bb64d0a422b873c1cbfe
BLAKE2b-256 00c3e392c1863eeb67a3a0b3acf3be5076254d24b93603c426e036942cbcb761

See more details on using hashes here.

File details

Details for the file dpdfnet-0.3.0-py3-none-any.whl.

File metadata

  • Download URL: dpdfnet-0.3.0-py3-none-any.whl
  • Upload date:
  • Size: 17.8 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.14

File hashes

Hashes for dpdfnet-0.3.0-py3-none-any.whl
Algorithm Hash digest
SHA256 f8042a2feed97b1b064d2310140fed7a7010c53dc6f4e6ae29a189c97f1556b4
MD5 bc33ea072e1b4cabf54fd2fd7b4a0661
BLAKE2b-256 0c5f87d4f6d4648ce73604604f4e86056bf001a1b867c82fbdbbece510388756

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page