fasr-enhancement-deepfilternet

英文文档地址: README_EN.md

DeepFilterNet 音频增强插件。离线 enhance() 和流式 push_chunk() 现在都 使用 deepfilternet-rs 包，由它通过 PyO3 将官方 DeepFilterNet Rust runtime 暴露给 Python。

安装

pip install fasr-enhancement-deepfilternet

stream_backend = "rust" 由独立的 deepfilternet-rs 包提供。使用 deepfilternet-rs 预编译 wheel 时，目标机器不需要 Rust。

注册模型

使用方式

from fasr.data import Waveform
from fasr_enhancement_deepfilternet import DeepFilterNetEnhancement

model = DeepFilterNetEnhancement(sample_rate=16000)
enhanced = model.enhance(Waveform(data=audio, sample_rate=16000))

流式使用：

from fasr.data import AudioChunk, Waveform
from fasr_enhancement_deepfilternet import DeepFilterNetEnhancement

model = DeepFilterNetEnhancement(sample_rate=16000, stream_backend="rust")
for enhanced_chunk in model.push_chunk(
    AudioChunk(
        stream_id="session-1",
        waveform=Waveform(data=chunk_audio, sample_rate=16000),
        is_last=False,
    )
):
    ...

Confection 配置

[enhancement_model]
@enhancement_models = "deepfilternet"
sample_rate = 16000
post_filter = true
compensate_delay = true
attenuation_limit_db = 100.0
stream_backend = "rust"
stream_log_level = "warn"
post_filter_beta = 0.0

参数

Realtime 状态

当前插件支持一种流式 push_chunk() backend：

• rust：使用 deepfilternet-rs，它通过 PyO3 封装官方 DeepFilterNet Rust DfTract runtime。

音频会在内部重采样到 backend 要求的采样率，处理完成后再重采样回 sample_rate 供下游 fasr VAD/ASR 使用。现在 batch 与 realtime 共用同一条 runtime 路径， 两种模式下的增强行为会保持一致。